km1lo commited on
Commit
41a8d54
·
verified ·
1 Parent(s): 83905f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -187,11 +187,12 @@ Redacta todo en máximo 300 palabras.
187
  # Generar salida
188
  generated_ids = model.generate(
189
  encoded,
190
- max_new_tokens=1000,
191
  do_sample=False,
192
- temperature=0.2
 
193
  )
194
-
195
  # Decodificar
196
  decoded = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
197
  auto_completo = decoded[0].strip()
 
187
  # Generar salida
188
  generated_ids = model.generate(
189
  encoded,
190
+ max_new_tokens=2000,
191
  do_sample=False,
192
+ temperature=0.2,
193
+ eos_token_id=tokenizer.eos_token_id
194
  )
195
+
196
  # Decodificar
197
  decoded = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
198
  auto_completo = decoded[0].strip()