Fix: make gpu selection respect the runner name capitalization (#142)

gpu-mode · Jan 19, 2025 · 9685dc5 · 9685dc5
1 parent 5e773dc
commit 9685dc5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/discord-cluster-manager/consts.py b/src/discord-cluster-manager/consts.py
@@ -25,7 +25,7 @@ class ModalGPU(Enum):
     H100 = "H100"
 
 
-GPU_SELECTION = {"Modal": ModalGPU, "Github": GitHubGPU}
+GPU_SELECTION = {"Modal": ModalGPU, "GitHub": GitHubGPU}
 
 
 class ExitCode(IntEnum):