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
This commit is contained in:
Shihao Liang
2025-05-19 17:15:17 +08:00
committed by GitHub
parent 51f5ddea04
commit 119bef25e2
2 changed files with 0 additions and 3 deletions

View File

@@ -824,12 +824,10 @@ class UITARSAgent:
frequency_penalty=1, frequency_penalty=1,
max_tokens=self.max_tokens, max_tokens=self.max_tokens,
temperature=temperature, temperature=temperature,
top_k=top_k,
top_p=self.top_p top_p=self.top_p
) )
# print(response.choices[0].message.content) # print(response.choices[0].message.content)
prediction = response.choices[0].message.content.strip() prediction = response.choices[0].message.content.strip()
prediction = response[0]["prediction"].strip()
except Exception as e: except Exception as e:
print(f"Error when fetching response from client, with response: {response}") print(f"Error when fetching response from client, with response: {response}")
prediction = None prediction = None

View File

@@ -156,7 +156,6 @@ def test(args: argparse.Namespace, test_all_meta: dict) -> None:
} }
agent = UITARSAgent( agent = UITARSAgent(
model=args.model,
action_space=args.action_space, action_space=args.action_space,
observation_type=args.observation_type, observation_type=args.observation_type,
max_trajectory_length=args.max_trajectory_length, max_trajectory_length=args.max_trajectory_length,