Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -64,8 +64,10 @@ def generate(
|
|
64 |
top_p=top_p,
|
65 |
#top_k=top_k,
|
66 |
temperature=temperature,
|
|
|
|
|
67 |
#num_beams=1,
|
68 |
-
repetition_penalty=1.2,
|
69 |
)
|
70 |
t = Thread(target=model.generate, kwargs=generate_kwargs)
|
71 |
t.start()
|
|
|
64 |
top_p=top_p,
|
65 |
#top_k=top_k,
|
66 |
temperature=temperature,
|
67 |
+
eos_token_id=pipe.tokenizer.eos_token_id,
|
68 |
+
pad_token_id=pipe.tokenizer.pad_token_id,
|
69 |
#num_beams=1,
|
70 |
+
#repetition_penalty=1.2,
|
71 |
)
|
72 |
t = Thread(target=model.generate, kwargs=generate_kwargs)
|
73 |
t.start()
|