dgjx commited on
Commit
cf16b57
·
verified ·
1 Parent(s): 04ce9c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -36,6 +36,7 @@ The following SQL query best answers the question `{user_question}`:
36
  eos_token_id=tokenizer.eos_token_id,
37
  pad_token_id=tokenizer.eos_token_id,
38
  max_new_tokens=400,
 
39
  num_beams=1,
40
  )
41
  outputs = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
 
36
  eos_token_id=tokenizer.eos_token_id,
37
  pad_token_id=tokenizer.eos_token_id,
38
  max_new_tokens=400,
39
+ do_sample=False,
40
  num_beams=1,
41
  )
42
  outputs = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)