Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def haiku_generate():
|
|
25 |
pad_token_id=tokenizer.pad_token_id,
|
26 |
do_sample=True,
|
27 |
repetition_penalty=1.2,
|
28 |
-
temperature=1.
|
29 |
top_p=0.9
|
30 |
)
|
31 |
generated_tokens = outputs[0][prompt_len:]
|
|
|
25 |
pad_token_id=tokenizer.pad_token_id,
|
26 |
do_sample=True,
|
27 |
repetition_penalty=1.2,
|
28 |
+
temperature=1.0,
|
29 |
top_p=0.9
|
30 |
)
|
31 |
generated_tokens = outputs[0][prompt_len:]
|