mirror/computer
1
0
Fork 0
mirror of https://github.com/trycua/computer.git synced 2026-01-07 14:00:04 -06:00
Code Issues Packages Projects Releases Wiki Activity

added grounding+planning composed loop

Browse Source
This commit is contained in:
Dillon DuPont
2025-08-04 16:32:05 -04:00
parent d5564977f0
commit f87b8eaea5
10 changed files with 904 additions and 400 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
libs/python/agent/benchmarks/models/__init__.py
View File

@@ -1,4 +1,3 @@
from .base import ModelProtocol
from .gta1 import GTA1Model
__all__ = ["ModelProtocol", "GTA1Model"]
__all__ = ["ModelProtocol"]

8
libs/python/agent/benchmarks/utils.py
View File

@@ -21,7 +21,6 @@ import torch
# Add parent directory to path for imports
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from agent.agent import ComputerAgent
from models import GTA1Model
from models.base import ModelProtocol
def get_gpu_memory() -> List[int]:
@@ -82,13 +81,16 @@ def get_available_models() -> List[Union[str, ModelProtocol]]:
"""
local_provider = "huggingface-local/" # Options: huggingface-local/ or mlx/
# from models.gta1 import GTA1Model
models = [
# === ComputerAgent model strings ===
f"{local_provider}HelloKKMe/GTA1-7B",
# f"{local_provider}HelloKKMe/GTA1-7B",
# f"{local_provider}HelloKKMe/GTA1-32B",
"openai/computer-use-preview+openai/gpt-4o-mini"
# === Reference model classes ===
GTA1Model("HelloKKMe/GTA1-7B"),
# GTA1Model("HelloKKMe/GTA1-7B"),
# GTA1Model("HelloKKMe/GTA1-32B"),
]