use alternate math-hard repo

This commit is contained in:
Wing Lian
2025-01-13 08:46:35 -05:00
parent fd91de3ea6
commit ee20600b9a
3 changed files with 13 additions and 1 deletions

View File

@@ -6,7 +6,7 @@ volumes:
secrets:
- HF_TOKEN
- WANDB_API_KEY
branch:
branch: cli-cloud-modal
gpu: h100
gpu_count: 1
memory: 128

11
lm_eval-kd.yaml Normal file
View File

@@ -0,0 +1,11 @@
lm_eval_model: axolotl-ai-co/numina-8b-ep1-exp1
lm_eval_tasks:
- leaderboard_math_hard
lm_eval_batch_size: 64
apply_chat_template: false
wandb_project: numina-kd-experiment
wandb_entity: axolotl-ai
bf16: true
flash_attention: true
output_dir: ./outputs/model-evals-out

View File

@@ -99,6 +99,7 @@ class ModalCloud(Cloud):
# Random id for cache busting of branch commits
f"RUN echo '{str(randint(0, 1000000))}'", # nosec B311
f"RUN cd /workspace/axolotl && git fetch && git checkout {self.config.branch}",
"RUN cd /workspace/ && git clone https://github.com/winglian/lm-evaluation-harness.git && cd lm-evaluation-harness && pip install -e .[math]",
]
)