Spaces:
Paused
Paused
Update inference.py
Browse files- inference.py +2 -2
inference.py
CHANGED
|
@@ -31,8 +31,8 @@ def process_document(file_path):
|
|
| 31 |
output = model.generate(
|
| 32 |
**inputs,
|
| 33 |
max_new_tokens=1500,
|
| 34 |
-
do_sample=False,
|
| 35 |
-
temperature=0.1
|
| 36 |
)
|
| 37 |
|
| 38 |
generated_ids = output[0][inputs.input_ids.shape[-1]:]
|
|
|
|
| 31 |
output = model.generate(
|
| 32 |
**inputs,
|
| 33 |
max_new_tokens=1500,
|
| 34 |
+
do_sample=False, # if it is true there will be extra text with output
|
| 35 |
+
# temperature=0.1 # temp is not required
|
| 36 |
)
|
| 37 |
|
| 38 |
generated_ids = output[0][inputs.input_ids.shape[-1]:]
|