File tree Expand file tree Collapse file tree 1 file changed +8
-3
lines changed
Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Original file line number Diff line number Diff line change 44 workflow_dispatch :
55 inputs :
66 instance_type :
7- description : ' Lambda instance type'
8- default : ' gpu_1x_a10'
7+ description : ' Lambda instance type(s), comma-separated for fallback'
8+ default : ' gpu_1x_a10,gpu_1x_a100,gpu_1x_h100_pcie'
9+ type : string
10+ region :
11+ description : ' Lambda region(s), comma-separated for fallback'
12+ default : ' us-south-1,us-west-1,us-east-1'
913 type : string
1014 epochs :
1115 description : ' Number of training epochs'
2428 LAMBDA_API_KEY : ${{ secrets.LAMBDA_API_KEY }}
2529 LAMBDA_SSH_PRIVATE_KEY : ${{ secrets.LAMBDA_SSH_PRIVATE_KEY }}
2630 with :
27- instance_type : ${{ inputs.instance_type || 'gpu_1x_a10' }}
31+ instance_type : ${{ inputs.instance_type || 'gpu_1x_a10,gpu_1x_a100,gpu_1x_h100_pcie' }}
32+ region : ${{ inputs.region || 'us-south-1,us-west-1,us-east-1' }}
2833 debug : ${{ inputs.debug || 'false' }}
2934
3035 gpu-test :
You can’t perform that action at this time.
0 commit comments