diff --git a/computer_use_demo/gui_agent/planner/local_vlm_planner.py b/computer_use_demo/gui_agent/planner/local_vlm_planner.py index e4a2463..330e2d0 100644 --- a/computer_use_demo/gui_agent/planner/local_vlm_planner.py +++ b/computer_use_demo/gui_agent/planner/local_vlm_planner.py @@ -25,7 +25,7 @@ """ MODEL_TO_HF_PATH = { - "qwen-vl-7b-instruct": "Qwen/Qwen2-VL-7B-Instruct", + "qwen2-vl-7b-instruct": "Qwen/Qwen2-VL-7B-Instruct", "qwen2-vl-2b-instruct": "Qwen/Qwen2-VL-2B-Instruct", "qwen2.5-vl-3b-instruct": "Qwen/Qwen2.5-VL-3B-Instruct", "qwen2.5-vl-7b-instruct": "Qwen/Qwen2.5-VL-7B-Instruct", @@ -224,4 +224,4 @@ def _message_filter_callback(messages): except Exception as e: print("[_message_filter_callback]: error", e) - return filtered_list \ No newline at end of file + return filtered_list diff --git a/computer_use_demo/loop.py b/computer_use_demo/loop.py index 5a511d9..264ac5d 100644 --- a/computer_use_demo/loop.py +++ b/computer_use_demo/loop.py @@ -134,7 +134,6 @@ def sampling_loop_sync( model=planner_model, provider=planner_provider, system_prompt_suffix=system_prompt_suffix, - api_key=api_key, api_response_callback=api_response_callback, selected_screen=selected_screen, output_callback=output_callback,