Spaces:
Running
Running
Update app.py
Browse files
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)
|