minor bug fix

socialfoundations · Jun 27, 2024 · f0ef464 · f0ef464
1 parent a7d7bad
commit f0ef464
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/folktexts/cli/eval_feature_importance.py b/folktexts/cli/eval_feature_importance.py
@@ -131,14 +131,14 @@ def main():
     # Set logging level
     logging.getLogger().setLevel(logging.INFO)
 
-    # Load model and tokenizer
+    # Resolve model folder path
     model_folder_path = Path(get_model_folder_path(model_name=args.model, root_dir=args.models_dir))
-    if model_folder_path.exists() and model_folder_path.is_dir():
-        logging.info(f"Loading model from {model_folder_path.as_posix()}")
-        model, tokenizer = load_model_tokenizer(model_folder_path)
-    else:
-        logging.info(f"Loading model from {Path(args.model).resolve().as_posix()}")
-        model, tokenizer = load_model_tokenizer(args.model)
+    if not model_folder_path.exists() or not model_folder_path.is_dir():
+        model_folder_path = Path(args.model).resolve()
+
+    # Load model and tokenizer
+    logging.info(f"Loading model from {model_folder_path.as_posix()}")
+    model, tokenizer = load_model_tokenizer(model_folder_path)
 
     # Set-up results directory
     results_dir = Path(args.results_dir) / Path(model_folder_path).name