From b3dae7fd4f2d521df7fd287682dd8e15ecffd30f Mon Sep 17 00:00:00 2001 From: AidanAbd Date: Sat, 14 Dec 2024 10:46:19 -0800 Subject: [PATCH] fix: incorrect gpu count --- src/latch/resources/tasks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/latch/resources/tasks.py b/src/latch/resources/tasks.py index ecfbfbf8..cf27314b 100644 --- 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"""