yashppawar commited on
Commit
4023331
·
verified ·
1 Parent(s): 9fcb15b

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- LOCAL_IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME", "postmortem_env-env:latest")
 
 
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
- env = await PostmortemEnv.from_docker_image(LOCAL_IMAGE_NAME)
 
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)