Update run_eval.py

allenai · Oct 30, 2023 · 39cf70b · 39cf70b
1 parent ddda985
commit 39cf70b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/eval/webshop/run_eval.py b/eval/webshop/run_eval.py
@@ -150,9 +150,9 @@ def lumos_iterative(args):
         if args.model_name_or_path:
             print("Loading model and tokenizer...")
             if i % 2 == 0:
-                model_name_or_path = os.path.join(args.model_name_or_path, "unified_plan_llama-2-7b")
+                model_name_or_path = os.path.join(args.model_name_or_path, "lumos_unified_plan_iterative")
             else:
-                model_name_or_path = os.path.join(args.model_name_or_path, "unified_ground_llama-2-7b")
+                model_name_or_path = os.path.join(args.model_name_or_path, "lumos_unified_ground_iterative")
 
             model, tokenizer = load_hf_lm_and_tokenizer(
                 model_name_or_path=model_name_or_path, 
@@ -226,4 +226,4 @@ def lumos_iterative(args):
     previous_actions = list()
     args.env = gym.make('WebAgentTextEnv-v0', observation_mode='text', num_products=DEBUG_PROD_SIZE)
 
-    lumos_iterative(args)
+    lumos_iterative(args)