Update extract_feature_print.py
Browse files- 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"]:
|