HusainHG commited on
Commit
506a1dc
·
verified ·
1 Parent(s): 7f5c9f2

Upload 9 files

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -119,16 +119,12 @@ def generate():
119
  sys.stdout.flush()
120
  generation_start = time.time()
121
 
122
- # Clamp temperature to avoid numerical issues
123
- temperature = max(0.5, min(temperature, 1.5))
124
-
125
  with torch.no_grad():
126
  outputs = model.generate(
127
  **inputs,
128
  max_new_tokens=max_new_tokens,
129
- temperature=temperature,
130
- top_p=top_p,
131
- do_sample=True,
132
  pad_token_id=tokenizer.eos_token_id,
133
  eos_token_id=tokenizer.eos_token_id,
134
  repetition_penalty=1.1,
 
119
  sys.stdout.flush()
120
  generation_start = time.time()
121
 
122
+ # Use greedy decoding to avoid temperature/sampling issues
 
 
123
  with torch.no_grad():
124
  outputs = model.generate(
125
  **inputs,
126
  max_new_tokens=max_new_tokens,
127
+ do_sample=False, # Use greedy decoding instead of sampling
 
 
128
  pad_token_id=tokenizer.eos_token_id,
129
  eos_token_id=tokenizer.eos_token_id,
130
  repetition_penalty=1.1,