Spaces:
Sleeping
Sleeping
Merge branch 'main' into last
Browse files- 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",
|
| 69 |
-
HF_TOKEN = os.environ.get("HF_TOKEN"
|
| 70 |
-
|
| 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)
|