fix: incorrect gpu count

latchbio · Dec 14, 2024 · b3dae7f · b3dae7f
1 parent 5a7950d
commit b3dae7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/latch/resources/tasks.py b/src/latch/resources/tasks.py
@@ -702,7 +702,7 @@ def _get_l40s_pod(instance_type: str, cpu: int, memory_gib: int, gpus: int) -> P
 
 g6e_16xlarge_task = functools.partial(
     task,
-    task_config=_get_l40s_pod("g6e-16xlarge", cpu=64, memory_gib=512, gpus=4)
+    task_config=_get_l40s_pod("g6e-16xlarge", cpu=64, memory_gib=512, gpus=1)
 )
 """64 vCPUs, 512 GiB RAM, 4 L40s GPUs"""