Pj12 commited on
Commit
0cc5fb5
·
verified ·
1 Parent(s): e744b4f

Update extract_feature_print.py

Browse files
Files changed (1) hide show
  1. extract_feature_print.py +1 -1
extract_feature_print.py CHANGED
@@ -177,7 +177,7 @@ def main():
177
  sys.exit(1)
178
 
179
  # Load model
180
- model_config = model_configs.get(model_name, model_configs[resolved_model_name])
181
  model_dict = model_config["load_model"](model_path, config.device, config.is_half)
182
  model = model_dict["model"]
183
  additional_configs = model_dict.get("saved_cfg")
 
177
  sys.exit(1)
178
 
179
  # Load model
180
+ model_config = model_configs.get(model_name, model_configs[model_name])
181
  model_dict = model_config["load_model"](model_path, config.device, config.is_half)
182
  model = model_dict["model"]
183
  additional_configs = model_dict.get("saved_cfg")