Pj12 commited on
Commit
70d2c50
·
verified ·
1 Parent(s): a62b9a2

Update extract_feature_print.py

Browse files
Files changed (1) hide show
  1. extract_feature_print.py +1 -4
extract_feature_print.py CHANGED
@@ -238,10 +238,7 @@ if os.access(model_path, os.F_OK) == False:
238
  % model_path
239
  )
240
  exit(0)
241
- models, saved_cfg, task = checkpoint_utils.load_model_ensemble_and_task(
242
- [model_path],
243
- suffix="",
244
- )
245
  if Custom_Embed == False:
246
  model = models[0]
247
  if device not in ["mps", "cpu"]:
 
238
  % model_path
239
  )
240
  exit(0)
241
+ models, saved_cfg, task = checkpoint_utils.load_model_ensemble_and_task([model_path])
 
 
 
242
  if Custom_Embed == False:
243
  model = models[0]
244
  if device not in ["mps", "cpu"]: