Tonic commited on
Commit
d1caca2
1 Parent(s): cad6f82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ Score 5: {score5_description}
37
 
38
  input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
39
 
40
- outputs = model.generate(input_ids, sample=True, temperature=1.0, top_p=0.9, max_new_tokens=650, repetition_penalty=1.03)
41
  result = tokenizer.decode(outputs[0])
42
 
43
  return result
 
37
 
38
  input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
39
 
40
+ outputs = model.generate(input_ids, do_sample=True, temperature=1.0, top_p=0.9, max_new_tokens=650, repetition_penalty=1.03)
41
  result = tokenizer.decode(outputs[0])
42
 
43
  return result