Upload utils.py

utils.py

@@ -8,7 +8,12 @@ def get_status(model_name: str):
     return client.get_model_status(model_name)
 
 def is_loadable(model_name: str, force_gpu: bool = False):
-    status = get_status(model_name)
+    try:
+        status = get_status(model_name)
+    except Exception as e:
+        print(e)
+        print(f"Couldn't load {model_name}.")
+        return False
     gpu_state = isinstance(status.compute_type, dict) and "gpu" in status.compute_type.keys()
     if status is None or status.state not in ["Loadable", "Loaded"] or (force_gpu and not gpu_state):
         print(f"Couldn't load {model_name}. Model state:'{status.state}', GPU:{gpu_state}")