Spaces:
Running
on
Zero
Running
on
Zero
Update model_manager.py
Browse files- model_manager.py +1 -1
model_manager.py
CHANGED
@@ -257,7 +257,7 @@ class ModelManager:
|
|
257 |
if "llama" in self.model_path.lower():
|
258 |
params.update({
|
259 |
"max_new_tokens": 600,
|
260 |
-
"temperature": 0.
|
261 |
"top_p": 0.75,
|
262 |
"repetition_penalty": 1.5,
|
263 |
"num_beams": 5,
|
|
|
257 |
if "llama" in self.model_path.lower():
|
258 |
params.update({
|
259 |
"max_new_tokens": 600,
|
260 |
+
"temperature": 0.35, # not too big
|
261 |
"top_p": 0.75,
|
262 |
"repetition_penalty": 1.5,
|
263 |
"num_beams": 5,
|