credent007 commited on
Commit
e7f2cb2
·
verified ·
1 Parent(s): 81767fa

Update inference.py

Browse files
Files changed (1) hide show
  1. 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]:]