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", diff --git a/src/autotrain/backends/base.py b/src/autotrain/backends/base.py index 187087b645..e9d34e69ff 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-l40sx4": "l40sx4", + "spaces-l40sx8": "l40sx8", "spaces-a10g-largex2": "a10g-largex2", "spaces-a10g-largex4": "a10g-largex4", # ngc