Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions deploy/modal/src/eval/evalscope.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import modal


app = modal.App("openai_gpt_oss_trl")
app = modal.App("openai_gpt_oss_evalscope")
IMAGE_GPU = os.getenv("IMAGE_GPU", None)
img = (
# https://catalog.ngc.nvidia.com/orgs/nvidia/containers/cuda/tags
Expand Down Expand Up @@ -42,7 +42,7 @@
model_path = os.path.join(HF_MODEL_DIR, MODEL_PATH)
model_name = MODEL_PATH.split("/")[-1]
eval_out_dir = os.path.join(EVAL_OUTPUT_DIR, "evalscope", MODEL_PATH.split("/")[-1])

os.makedirs(eval_out_dir, exist_ok=True)

@app.function(
gpu=IMAGE_GPU,
Expand Down
5 changes: 3 additions & 2 deletions deploy/modal/src/eval/lighteval.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import modal


app = modal.App("openai_gpt_oss_trl")
app = modal.App("openai_gpt_oss_lighteval")
IMAGE_GPU = os.getenv("IMAGE_GPU", None)
img = (
# https://catalog.ngc.nvidia.com/orgs/nvidia/containers/cuda/tags
Expand Down Expand Up @@ -55,7 +55,8 @@

MODEL_PATH = os.getenv("LLM_MODEL", "openai/gpt-oss-20b")
model_path = os.path.join(HF_MODEL_DIR, MODEL_PATH)
eval_out_dir = os.path.join(EVAL_OUTPUT_DIR, MODEL_PATH.split("/")[-1])
eval_out_dir = os.path.join(EVAL_OUTPUT_DIR, "lighteval", MODEL_PATH.split("/")[-1])
os.makedirs(eval_out_dir, exist_ok=True)


@app.function(
Expand Down
Loading