dh-mc commited on
Commit
52752a1
1 Parent(s): 5bda5b6

Update llm_loader.py

Browse files
Files changed (1) hide show
  1. app_modules/llm_loader.py +1 -0
app_modules/llm_loader.py CHANGED
@@ -219,6 +219,7 @@ class LLMLoader:
219
  temperature=0.01,
220
  repetition_penalty=1.115,
221
  callbacks=callbacks,
 
222
  streaming=True,
223
  )
224
  elif self.llm_model_type.startswith("huggingface"):
 
219
  temperature=0.01,
220
  repetition_penalty=1.115,
221
  callbacks=callbacks,
222
+ timeout=600,
223
  streaming=True,
224
  )
225
  elif self.llm_model_type.startswith("huggingface"):