diff --git a/pipeline/cloud/compute_requirements.py b/pipeline/cloud/compute_requirements.py index 8f95b071..9eec048b 100644 --- a/pipeline/cloud/compute_requirements.py +++ b/pipeline/cloud/compute_requirements.py @@ -19,6 +19,9 @@ class Accelerator(str, Enum): nvidia_a100_80gb_40gb: str = "nvidia_a100_80gb_40gb" nvidia_a10: str = "nvidia_a10" + nvidia_a10_12gb: str = "nvidia_a10_12gb" + nvidia_a10_8gb: str = "nvidia_a10_8gb" + nvidia_a10_4gb: str = "nvidia_a10_4gb" @classmethod def from_str(cls, accelerator: str) -> "Accelerator": @@ -71,4 +74,7 @@ def valid_accelerator_config(cls, accelerators: list["Accelerator"]): Accelerator.nvidia_a100_80gb_20gb, Accelerator.nvidia_a100_80gb_40gb, Accelerator.nvidia_a10, + Accelerator.nvidia_a10_12gb, + Accelerator.nvidia_a10_8gb, + Accelerator.nvidia_a10_4gb, ] diff --git a/pyproject.toml b/pyproject.toml index 45cb3619..0d834f2d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "pipeline-ai" -version = "2.1.2" +version = "2.1.3" description = "Pipelines for machine learning workloads." authors = [ "Paul Hetherington ",