Spaces:
Running
Running
admin
commited on
Commit
·
2b6113a
1
Parent(s):
f1b22d5
fix assign_model.lower()
Browse files
utils.py
CHANGED
@@ -35,7 +35,7 @@ def get_modelist(model_dir=MODEL_DIR, assign_model=""):
|
|
35 |
for file in files:
|
36 |
if file.endswith(".pt"):
|
37 |
model = os.path.basename(file)[:-3]
|
38 |
-
if assign_model and assign_model in model:
|
39 |
pt_files.append(model)
|
40 |
else:
|
41 |
pt_files.insert(0, model)
|
|
|
35 |
for file in files:
|
36 |
if file.endswith(".pt"):
|
37 |
model = os.path.basename(file)[:-3]
|
38 |
+
if assign_model and assign_model.lower() in model:
|
39 |
pt_files.append(model)
|
40 |
else:
|
41 |
pt_files.insert(0, model)
|