Add another variant "nvidia-h100-nvl"

huggingface · Dec 13, 2024 · 985bd6a · 985bd6a
1 parent 3da49f3
commit 985bd6a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/launcher/src/main.rs b/launcher/src/main.rs
@@ -1652,6 +1652,7 @@ impl From<&str> for Gpu {
             "nvidia-l40s" => Gpu::L40S,
             "nvidia-a10g" => Gpu::A10G,
             "nvidia-h100-80gb-hbm3" => Gpu::H100,
+            "nvidia-h100-nvl" => Gpu::H100,
             "nvidia-h100" => Gpu::H100,
             "nvidia-a100-sxm4-80gb" => Gpu::A100,
             "nvidia-a100-80gb-pcie" => Gpu::A100,