Tusharp2006 commited on
Commit
7da2612
Β·
unverified Β·
2 Parent(s): 3da2c872b43ff4

Merge branch 'main' into last

Browse files
Files changed (1) hide show
  1. inference.py +4 -4
inference.py CHANGED
@@ -65,9 +65,9 @@ except ImportError:
65
 
66
  # ── Env-var config (checklist-specified names) ────────────────────────────────
67
  API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
68
- MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o-mini") # or gpt-4o
69
- HF_TOKEN = os.environ.get("HF_TOKEN", "")
70
- OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY", "no-key-set")
71
 
72
  # ── Task registry ─────────────────────────────────────────────────────────────
73
  _TASKS: Dict[str, Dict[str, Any]] = {
@@ -386,4 +386,4 @@ if __name__ == "__main__":
386
  args = p.parse_args()
387
 
388
  task_list = [args.task] if args.task else ["easy", "medium", "hard"]
389
- run_baseline(tasks=task_list, num_episodes=args.n, seed_offset=args.seed)
 
65
 
66
  # ── Env-var config (checklist-specified names) ────────────────────────────────
67
  API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
68
+ MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o")
69
+ HF_TOKEN = os.environ.get("HF_TOKEN")
70
+ _API_KEY = HF_TOKEN or os.environ.get("OPENAI_API_KEY", "no-key-set")
71
 
72
  # ── Task registry ─────────────────────────────────────────────────────────────
73
  _TASKS: Dict[str, Dict[str, Any]] = {
 
386
  args = p.parse_args()
387
 
388
  task_list = [args.task] if args.task else ["easy", "medium", "hard"]
389
+ run_baseline(tasks=task_list, num_episodes=args.n, seed_offset=args.seed)