TymaaHammouda commited on
Commit
5f20637
·
verified ·
1 Parent(s): 360cd0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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",