Spaces:

yasu-oh
/

quantized_model_memory_estimator

Sleeping

yasu-oh commited on May 4

Commit

5df4a84

1 Parent(s): 47f79b3

judg

Files changed (1) hide show

app.py CHANGED Viewed

@@ -58,7 +58,7 @@ def is_architecture_supported(architecture: str) -> Optional[bool]:
             for name in match.replace('\n', '').replace('"', '').split(',')
             if name.strip()
         ]
-        return any(architecture in entry for entry in model_names)
     except:
         return None

             for name in match.replace('\n', '').replace('"', '').split(',')
             if name.strip()
         ]
+        return any(architecture == entry for entry in model_names)
     except:
         return None