use alternate math-hard repo
This commit is contained in:
@@ -6,7 +6,7 @@ volumes:
|
|||||||
secrets:
|
secrets:
|
||||||
- HF_TOKEN
|
- HF_TOKEN
|
||||||
- WANDB_API_KEY
|
- WANDB_API_KEY
|
||||||
branch:
|
branch: cli-cloud-modal
|
||||||
gpu: h100
|
gpu: h100
|
||||||
gpu_count: 1
|
gpu_count: 1
|
||||||
memory: 128
|
memory: 128
|
||||||
|
|||||||
11
lm_eval-kd.yaml
Normal file
11
lm_eval-kd.yaml
Normal 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
|
||||||
@@ -99,6 +99,7 @@ class ModalCloud(Cloud):
|
|||||||
# Random id for cache busting of branch commits
|
# Random id for cache busting of branch commits
|
||||||
f"RUN echo '{str(randint(0, 1000000))}'", # nosec B311
|
f"RUN echo '{str(randint(0, 1000000))}'", # nosec B311
|
||||||
f"RUN cd /workspace/axolotl && git fetch && git checkout {self.config.branch}",
|
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]",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user