Skip to content

Commit 1b924c9

Browse files
fix: allow more cuda versions; use proper name of the MODEL_NAME (#28)
1 parent 78dcd24 commit 1b924c9

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.runpod/hub.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
"containerDiskInGb": 100,
1010
"gpuIds": "ADA_24",
1111
"gpuCount": 1,
12-
"allowedCudaVersions": ["12.7", "12.6", "12.5", "12.4"],
12+
"allowedCudaVersions": ["12.9", "12.8", "12.7", "12.6"],
1313
"env": [
1414
{
1515
"key": "MODEL_NAME",

.runpod/tests.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,10 @@
2020
"value": "true"
2121
},
2222
{
23-
"key": "MODEL_PATH",
23+
"key": "MODEL_NAME",
2424
"value": "HuggingFaceTB/SmolLM2-1.7B-Instruct"
2525
}
2626
],
27-
"allowedCudaVersions": ["12.7", "12.6", "12.5", "12.4"]
27+
"allowedCudaVersions": ["12.9", "12.8", "12.7", "12.6"]
2828
}
2929
}

0 commit comments

Comments
 (0)