Upload 9 files
Browse files
app.py
CHANGED
|
@@ -119,16 +119,12 @@ def generate():
|
|
| 119 |
sys.stdout.flush()
|
| 120 |
generation_start = time.time()
|
| 121 |
|
| 122 |
-
#
|
| 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 |
-
|
| 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,
|