Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def haiku_generate():
|
|
20 |
with torch.no_grad():
|
21 |
outputs = model.generate(
|
22 |
**inputs,
|
23 |
-
max_new_tokens=
|
24 |
eos_token_id=tokenizer.eos_token_id, # 終わりを明示!
|
25 |
pad_token_id=tokenizer.pad_token_id,
|
26 |
do_sample=True,
|
|
|
20 |
with torch.no_grad():
|
21 |
outputs = model.generate(
|
22 |
**inputs,
|
23 |
+
max_new_tokens=20,
|
24 |
eos_token_id=tokenizer.eos_token_id, # 終わりを明示!
|
25 |
pad_token_id=tokenizer.pad_token_id,
|
26 |
do_sample=True,
|