Spaces:

r3gm
/

DiffuseCraft

Running on Zero

r3gm commited on Dec 19, 2024

Commit

5c4ab14

verified ·

1 Parent(s): 8f4bb65

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -300,7 +300,7 @@ def get_model_type(repo_id: str):
     default = "SD 1.5"
     try:
         if os.path.exists(repo_id):
-            tag, _, _ = checkpoint_model_type(repo_id)
             return DIFFUSECRAFT_CHECKPOINT_NAME[tag]
         else:
             model = api.model_info(repo_id=repo_id, timeout=5.0)

     default = "SD 1.5"
     try:
         if os.path.exists(repo_id):
+            tag, _, _, _ = checkpoint_model_type(repo_id)
             return DIFFUSECRAFT_CHECKPOINT_NAME[tag]
         else:
             model = api.model_info(repo_id=repo_id, timeout=5.0)