Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -59,8 +59,8 @@ class HfModelWrapper:
|
|
59 |
"top_k": 50,
|
60 |
"repetition_penalty": 1.1,
|
61 |
"do_sample": True,
|
62 |
-
"eos_token_id": tokenizer.eos_token_id,
|
63 |
-
"pad_token_id": tokenizer.pad_token_id,
|
64 |
}
|
65 |
|
66 |
thread = Thread(target=self.model.generate, kwargs=gen_kwargs)
|
|
|
59 |
"top_k": 50,
|
60 |
"repetition_penalty": 1.1,
|
61 |
"do_sample": True,
|
62 |
+
"eos_token_id": self.tokenizer.eos_token_id,
|
63 |
+
"pad_token_id": self.tokenizer.pad_token_id,
|
64 |
}
|
65 |
|
66 |
thread = Thread(target=self.model.generate, kwargs=gen_kwargs)
|