Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -55,8 +55,9 @@ class HfModelWrapper:
|
|
55 |
"pad_token_id": self.tokenizer.eos_token_id,
|
56 |
"max_length": 32768,
|
57 |
"temperature": 0.1,
|
58 |
-
"top_p": 0.
|
59 |
-
"repetition_penalty": 1.1,
|
|
|
60 |
}
|
61 |
|
62 |
thread = Thread(target=self.model.generate, kwargs=gen_kwargs)
|
|
|
55 |
"pad_token_id": self.tokenizer.eos_token_id,
|
56 |
"max_length": 32768,
|
57 |
"temperature": 0.1,
|
58 |
+
"top_p": 0.9,
|
59 |
+
#"repetition_penalty": 1.1,
|
60 |
+
"min_length": 5,
|
61 |
}
|
62 |
|
63 |
thread = Thread(target=self.model.generate, kwargs=gen_kwargs)
|