From 119bef25e22f3905032d19b7aefcd133a5701070 Mon Sep 17 00:00:00 2001
From: Shihao Liang <1017512024@qq.com>
Date: Mon, 19 May 2025 17:15:17 +0800
Subject: [PATCH] Dev/uitars 15 (#194)

* debug uitars1.0, add uitars1.5

* update pyautogui parser

* modify function name

* update parser

* update prompt

* FIX: bug in ui tars
---
 mm_agents/uitars_agent.py | 2 --
 run_uitars.py             | 1 -
 2 files changed, 3 deletions(-)

diff --git a/mm_agents/uitars_agent.py b/mm_agents/uitars_agent.py
index 1c40f76..245a5c3 100644
--- a/mm_agents/uitars_agent.py
+++ b/mm_agents/uitars_agent.py
@@ -824,12 +824,10 @@ class UITARSAgent:
                     frequency_penalty=1,
                     max_tokens=self.max_tokens,
                     temperature=temperature,
-                    top_k=top_k,
                     top_p=self.top_p
                 )
                 # print(response.choices[0].message.content)
                 prediction = response.choices[0].message.content.strip()
-                prediction = response[0]["prediction"].strip()
             except Exception as e:
                 print(f"Error when fetching response from client, with response: {response}")
                 prediction = None
diff --git a/run_uitars.py b/run_uitars.py
index cf0701a..aa11246 100644
--- a/run_uitars.py
+++ b/run_uitars.py
@@ -156,7 +156,6 @@ def test(args: argparse.Namespace, test_all_meta: dict) -> None:
     }
 
     agent = UITARSAgent(
-        model=args.model,
         action_space=args.action_space,
         observation_type=args.observation_type,
         max_trajectory_length=args.max_trajectory_length,