From 5c226b600d5bb1207374abde30c8b4af89b43a04 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 8 Jan 2025 08:38:06 -0500 Subject: [PATCH] pr feedback --- src/axolotl/cli/cloud/modal_.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/axolotl/cli/cloud/modal_.py b/src/axolotl/cli/cloud/modal_.py index 8d497ddd4..412d56696 100644 --- a/src/axolotl/cli/cloud/modal_.py +++ b/src/axolotl/cli/cloud/modal_.py @@ -168,7 +168,7 @@ class ModalCloud(Cloud): def get_train_timeout(self): if self.config.timeout: return int(self.config.timeout) - return 60 * 60 * 24 # 30 hours + return 60 * 60 * 24 # 24 hours def get_train_gpu(self): # pylint: disable=too-many-return-statements count = self.config.gpu_count or 1 @@ -180,7 +180,7 @@ class ModalCloud(Cloud): return modal.gpu.A100(count=count, size="40GB") if family == "a100-80gb": return modal.gpu.A100(count=count, size="80GB") - if family == "a10g": + if family in ["a10", "a10g"]: return modal.gpu.A10G(count=count) if family == "h100": return modal.gpu.H100(count=count)