File tree Expand file tree Collapse file tree 1 file changed +1
-21
lines changed
Expand file tree Collapse file tree 1 file changed +1
-21
lines changed Original file line number Diff line number Diff line change @@ -2,32 +2,13 @@ name: "BioNeMo Model Convergence Tests"
22
33on :
44 workflow_dispatch :
5- inputs :
6- branch :
7- description : " Branch to use"
8- required : true
9- default : " main"
10- type : string
11- model_config :
12- description : " Model configuration to use"
13- required : true
14- default : " esm2_accelerate"
15- type : choice
16- options :
17- - amplify_accelerate_te_fp8.yaml
18- - esm2_accelerate
19- - esm2_native_te_mfsdp
20- - esm2_native_te_nvfsdp_thd
21- - geneformer_native_te_nvfsdp_fp8
225
236jobs :
247 submit-lepton-jobs :
258 runs-on : ubuntu-latest
269 steps :
2710 - name : Checkout
2811 uses : actions/checkout@v4
29- with :
30- ref : ${{ inputs.branch }}
3112
3213 - name : Set up Python
3314 uses : actions/setup-python@v5
5233 LEP_LOGIN_CREDENTIALS : ${{ secrets.LEP_LOGIN_CREDENTIALS }}
5334 run : |
5435 lep login -c "$LEP_LOGIN_CREDENTIALS" || true
55- pwd
56- python ci/lepton/model_convergence/scripts/launch_job.py --config-name="recipes/${{ inputs.model_config }}" branch="${{ inputs.branch }}"
36+ python ci/lepton/model_convergence/scripts/launch_job.py --config-name "evo2_finetune_lora"
You can’t perform that action at this time.
0 commit comments