Add H100, K80 to NOS devices

autonomi-ai · Jan 25, 2024 · 20cf1aa · 20cf1aa
1 parent 728abba
commit 20cf1aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nos/constants.py b/nos/constants.py
@@ -41,4 +41,4 @@
 # Edit: apparently the device names generated by skypilot do not match those
 # that come out of the model spec (unclear where those come from?) right now
 # just need to run these one by one and figure out what string gets used and update.
-SKYPILOT_DEVICES = ["tesla-t4", "a100"]
+SKYPILOT_DEVICES = ["tesla-t4", "a100", "h100", "k80"]