Yntec commited on
Commit
759203c
β€’
1 Parent(s): 3a057ea

Update externalmod.py

Browse files
Files changed (1) hide show
  1. externalmod.py +1 -1
externalmod.py CHANGED
@@ -115,7 +115,7 @@ def from_model(model_name: str, hf_token: str | None, alias: str | None, **kwarg
115
 
116
  headers["X-Wait-For-Model"] = "true"
117
  client = huggingface_hub.InferenceClient(
118
- model=model_name, headers=headers, token=hf_token
119
  )
120
 
121
  # For tasks that are not yet supported by the InferenceClient
 
115
 
116
  headers["X-Wait-For-Model"] = "true"
117
  client = huggingface_hub.InferenceClient(
118
+ model=model_name, headers=headers, token=hf_token, timeout=120,
119
  )
120
 
121
  # For tasks that are not yet supported by the InferenceClient