admin commited on
Commit
2b6113a
·
1 Parent(s): f1b22d5

fix assign_model.lower()

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
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)