From 1d1f14ee9e38e456e0d5aab23ba07f13f871edae Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Wed, 11 Sep 2024 17:57:21 -0500 Subject: [PATCH 1/3] Add new L40S hardware flavors --- src/autotrain/app/api_routes.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/autotrain/app/api_routes.py b/src/autotrain/app/api_routes.py index cc666777fc..7ccd8c14ab 100644 --- a/src/autotrain/app/api_routes.py +++ b/src/autotrain/app/api_routes.py @@ -242,6 +242,9 @@ class APICreateProjectModel(BaseModel): "spaces-cpu-basic", "spaces-l4x1", "spaces-l4x4", + "spaces-l40sx1", + "spaces-l40sx4", + "spaces-l40sx8", "spaces-a10g-largex2", "spaces-a10g-largex4", # "local", From 0fba0970fba40a434110be9571fa2c854ee30b04 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Wed, 11 Sep 2024 18:00:07 -0500 Subject: [PATCH 2/3] add to base --- src/autotrain/backends/base.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/autotrain/backends/base.py b/src/autotrain/backends/base.py index 187087b645..01e08349ea 100644 --- a/src/autotrain/backends/base.py +++ b/src/autotrain/backends/base.py @@ -29,6 +29,9 @@ "spaces-cpu-basic": "cpu-basic", "spaces-l4x1": "l4x1", "spaces-l4x4": "l4x4", + "spaces-l40sx1": "l40sx1", + "spaces-l40sx1": "l40sx4", + "spaces-l40sx1": "l40sx8", "spaces-a10g-largex2": "a10g-largex2", "spaces-a10g-largex4": "a10g-largex4", # ngc From 70839dcd67d4efe0a84129a297b1d26b2a4e0e32 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Wed, 11 Sep 2024 18:00:29 -0500 Subject: [PATCH 3/3] fix --- src/autotrain/backends/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/autotrain/backends/base.py b/src/autotrain/backends/base.py index 01e08349ea..e9d34e69ff 100644 --- a/src/autotrain/backends/base.py +++ b/src/autotrain/backends/base.py @@ -30,8 +30,8 @@ "spaces-l4x1": "l4x1", "spaces-l4x4": "l4x4", "spaces-l40sx1": "l40sx1", - "spaces-l40sx1": "l40sx4", - "spaces-l40sx1": "l40sx8", + "spaces-l40sx4": "l40sx4", + "spaces-l40sx8": "l40sx8", "spaces-a10g-largex2": "a10g-largex2", "spaces-a10g-largex4": "a10g-largex4", # ngc