Merge pull request #159 from backend-developers-ltd/always_on

Added 'always_on.gpu-24gb' executor class
backend-developers-ltd · Jul 3, 2024 · fce7885 · fce7885
2 parents dbad485 + 02664f1
commit fce7885
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/compute_horde/compute_horde/executor_class.py b/compute_horde/compute_horde/executor_class.py
@@ -5,8 +5,8 @@
 
 class ExecutorClass(StrEnum):
     spin_up_4min__gpu_24gb = "spin_up-4min.gpu-24gb"
+    always_on__gpu_24gb = "always_on.gpu-24gb"
     # always_on__cpu_16c__ram_64gb = "always_on.cpu-16c.ram-64gb"
-    # always_on__gpu_24gb = "always_on.gpu-24gb"
     # always_on__gpu_80gb = "always_on.gpu-80gb"
     # always_on__gpu_24gb__docker_cached_facilitator = "always_on.gpu-24gb.docker_cached-facilitator"
 
@@ -33,16 +33,17 @@ class ExecutorClassSpec:
         gpu_vram_gb=24,
         spin_up_time=int(timedelta(minutes=4).total_seconds()),
     ),
+    ExecutorClass.always_on__gpu_24gb: ExecutorClassSpec(
+        description="always on, generic GPU machine",
+        has_gpu=True,
+        gpu_vram_gb=24,
+        spin_up_time=0,
+    ),
     # ExecutorClass.always_on__cpu_16c__ram_64gb: ExecutorClassSpec(
     #     cpu_cores=16,
     #     ram_gb=64,
     #     spin_up_time=0,
     # ),
-    # ExecutorClass.always_on__gpu_24gb: ExecutorClassSpec(
-    #     has_gpu=True,
-    #     gpu_vram_gb=24,
-    #     spin_up_time=0,
-    # ),
     # ExecutorClass.always_on__gpu_80gb: ExecutorClassSpec(
     #     has_gpu=True,
     #     gpu_vram_gb=80,