credent007 commited on
Commit
0eb7146
·
verified ·
1 Parent(s): 50546ef

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -30,7 +30,7 @@ def process_document(file_path):
30
 
31
  output = model.generate(
32
  **inputs,
33
- max_new_tokens=150,
34
  do_sample=False,
35
  temperature=0.1
36
  )
 
30
 
31
  output = model.generate(
32
  **inputs,
33
+ max_new_tokens=1500,
34
  do_sample=False,
35
  temperature=0.1
36
  )