Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,8 +10,9 @@ model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
|
|
| 10 |
|
| 11 |
@spaces.GPU(duration=120)
|
| 12 |
def predict(message, history, temperature, top_p):
|
|
|
|
| 13 |
if len(history) == 0:
|
| 14 |
-
history.append({"role": "
|
| 15 |
You are the Tootsie 8B advanced language model trained using Marin, a framework developed by Stanford's Center for Research on Foundation Models (CRFM).
|
| 16 |
|
| 17 |
Marin is a framework designed for training large language models in an entirely open fashion with a focus on legibility, scalability, and reproducibility.
|
|
|
|
| 10 |
|
| 11 |
@spaces.GPU(duration=120)
|
| 12 |
def predict(message, history, temperature, top_p):
|
| 13 |
+
print(history)
|
| 14 |
if len(history) == 0:
|
| 15 |
+
history.append({"role": "system", "content": """
|
| 16 |
You are the Tootsie 8B advanced language model trained using Marin, a framework developed by Stanford's Center for Research on Foundation Models (CRFM).
|
| 17 |
|
| 18 |
Marin is a framework designed for training large language models in an entirely open fashion with a focus on legibility, scalability, and reproducibility.
|