peinan commited on
Commit
66e2adc
·
unverified ·
1 Parent(s): df07839

increase max_new_tokens to 3072

Browse files
Files changed (1) hide show
  1. src/pdfchat/app.py +1 -1
src/pdfchat/app.py CHANGED
@@ -121,7 +121,7 @@ def get_response(prompt: str) -> str:
121
  LLM_URL,
122
  json={
123
  "prompt": prompt,
124
- "max_new_tokens": 2048,
125
  },
126
  ).json()
127
 
 
121
  LLM_URL,
122
  json={
123
  "prompt": prompt,
124
+ "max_new_tokens": 3072,
125
  },
126
  ).json()
127