Togethercompute

app.py

@@ -76,7 +76,9 @@ def calculate_memory(model_name:str, library:str, options:list, access_token:str
         library = check_has_model(e)
         if library != "unknown":
             raise gr.Error(f"Tried to load `{model_name}` with `{library}` but a possible model to load was not found inside the repo.")
-    
+    except ImportError:
+        # hacky way to check if it works with `trust_remote_code=False`
+        model = create_empty_model(model_name, library_name=library, trust_remote_code=False, access_token=access_token)
     total_size, largest_layer = calculate_maximum_sizes(model)
 
     data = []