MrDrmm commited on
Commit
9179d67
·
verified ·
1 Parent(s): 58499f8

Update multit2i.py

Browse files
Files changed (1) hide show
  1. multit2i.py +2 -2
multit2i.py CHANGED
@@ -35,11 +35,11 @@ def is_repo_name(s):
35
 
36
  def get_status(model_name: str):
37
  from huggingface_hub import InferenceClient
38
- client = InferenceClient(token=HF_TOKEN, timeout=50)
39
  return client.get_model_status(model_name)
40
 
41
 
42
- def is_loadable(model_name: str, force_gpu: bool = False):
43
  try:
44
  status = get_status(model_name)
45
  except Exception as e:
 
35
 
36
  def get_status(model_name: str):
37
  from huggingface_hub import InferenceClient
38
+ client = InferenceClient(token=HF_TOKEN, timeout=150)
39
  return client.get_model_status(model_name)
40
 
41
 
42
+ def is_loadable(model_name: str, force_gpu: bool = True):
43
  try:
44
  status = get_status(model_name)
45
  except Exception as e: