Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ def load_models():
|
|
| 52 |
|
| 53 |
app.state.deepseek_model = AutoModelForCausalLM.from_pretrained(
|
| 54 |
model_name,
|
| 55 |
-
max_tokens=10, # Limit to very few tokens
|
| 56 |
temperature=0.1, # Low temperature for more deterministic output
|
| 57 |
do_sample=False, # Disable sampling for consistent results
|
| 58 |
torch_dtype="auto",
|
|
|
|
| 52 |
|
| 53 |
app.state.deepseek_model = AutoModelForCausalLM.from_pretrained(
|
| 54 |
model_name,
|
| 55 |
+
# max_tokens=10, # Limit to very few tokens
|
| 56 |
temperature=0.1, # Low temperature for more deterministic output
|
| 57 |
do_sample=False, # Disable sampling for consistent results
|
| 58 |
torch_dtype="auto",
|