Spaces:
Runtime error
Runtime error
Update multit2i.py
Browse files- 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=
|
39 |
return client.get_model_status(model_name)
|
40 |
|
41 |
|
42 |
-
def is_loadable(model_name: str, force_gpu: bool =
|
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:
|