Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- inference.py +5 -2
inference.py
CHANGED
|
@@ -49,7 +49,9 @@ MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
|
|
| 49 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 50 |
|
| 51 |
# Optional — used only when launching the env from a local Docker image.
|
| 52 |
-
|
|
|
|
|
|
|
| 53 |
|
| 54 |
BENCHMARK = "postmortem_env"
|
| 55 |
|
|
@@ -270,7 +272,8 @@ async def main() -> None:
|
|
| 270 |
|
| 271 |
# Spin up one containerised env; run the 3 scenarios back-to-back
|
| 272 |
# by issuing 3 resets (the env rotates scenarios on each reset).
|
| 273 |
-
|
|
|
|
| 274 |
try:
|
| 275 |
for _ in range(3):
|
| 276 |
await run_one_task(client, env)
|
|
|
|
| 49 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 50 |
|
| 51 |
# Optional — used only when launching the env from a local Docker image.
|
| 52 |
+
# Per the submission checklist, defaults are set ONLY for API_BASE_URL and MODEL_NAME.
|
| 53 |
+
LOCAL_IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME")
|
| 54 |
+
_DEFAULT_IMAGE_FALLBACK = "postmortem_env-env:latest"
|
| 55 |
|
| 56 |
BENCHMARK = "postmortem_env"
|
| 57 |
|
|
|
|
| 272 |
|
| 273 |
# Spin up one containerised env; run the 3 scenarios back-to-back
|
| 274 |
# by issuing 3 resets (the env rotates scenarios on each reset).
|
| 275 |
+
image = LOCAL_IMAGE_NAME or _DEFAULT_IMAGE_FALLBACK
|
| 276 |
+
env = await PostmortemEnv.from_docker_image(image)
|
| 277 |
try:
|
| 278 |
for _ in range(3):
|
| 279 |
await run_one_task(client, env)
|