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.
|