Merge remote-tracking branch 'origin/main'

This commit is contained in:
Timothyxxx
2024-01-31 23:53:05 +08:00
27 changed files with 215 additions and 15 deletions

View File

@@ -1,4 +1,4 @@
import ctypes
#import ctypes
import datetime
import json
import logging
@@ -131,8 +131,10 @@ def main(example_class, example_id):
example = json.load(f)
example["snapshot"] = "exp_v5"
logger.info("TASK: %s/%s", example_class, example_id)
api_key = os.environ.get("OPENAI_API_KEY")
agent = GPT4v_Agent(api_key=api_key, model=gpt4_model, instruction=example['instruction'],
agent = GPT4v_Agent(api_key=api_key, model=gpt4_model, max_tokens=1000, instruction=example['instruction'],
action_space=action_space, exp="som")
# api_key = os.environ.get("GENAI_API_KEY")