Spaces:
Runtime error
Runtime error
Bug fix
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ def generate(
|
|
28 |
model, tok, text,
|
29 |
do_sample=True, max_length=10000, repetition_penalty=5.0,
|
30 |
top_k=5, top_p=0.95, temperature=1,
|
31 |
-
num_beams=
|
32 |
no_repeat_ngram_size=3
|
33 |
):
|
34 |
input_ids = tok.encode(text, return_tensors="pt").to(DEVICE)
|
|
|
28 |
model, tok, text,
|
29 |
do_sample=True, max_length=10000, repetition_penalty=5.0,
|
30 |
top_k=5, top_p=0.95, temperature=1,
|
31 |
+
num_beams=10,
|
32 |
no_repeat_ngram_size=3
|
33 |
):
|
34 |
input_ids = tok.encode(text, return_tensors="pt").to(DEVICE)
|