diff --git a/pyproject.toml b/pyproject.toml index d9ae1dd0b..728bfaa93 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "truss" -version = "0.18.23" +version = "0.18.25rc0" description = "A seamless bridge from model development to model delivery" authors = [ { name = "Pankaj Gupta", email = "no-reply@baseten.co" }, diff --git a/truss/base/truss_config.py b/truss/base/truss_config.py index f083493c4..3e2f5fbc1 100644 --- a/truss/base/truss_config.py +++ b/truss/base/truss_config.py @@ -93,6 +93,7 @@ class Accelerator(str, enum.Enum): B200 = "B200" L40S = "L40S" RTX_PRO_6000 = "RTX_PRO_6000" + RTX_PRO_6000_48GB_GFX = "RTX_PRO_6000_48GB_GFX" B300 = "B300" GB300 = "GB300" diff --git a/truss/contexts/docker_build_setup.py b/truss/contexts/docker_build_setup.py index 055be8ff6..e0b4f0472 100644 --- a/truss/contexts/docker_build_setup.py +++ b/truss/contexts/docker_build_setup.py @@ -63,6 +63,7 @@ def _fill_trt_llm_versions( Accelerator.V100: "turing-", Accelerator.B200: "blackwell-", Accelerator.RTX_PRO_6000: "blackwell-", + Accelerator.RTX_PRO_6000_48GB_GFX: "blackwell-", Accelerator.B300: "blackwell-", Accelerator.GB300: "blackwell-", None: "unsupported none please upgrade truss", diff --git a/truss/tests/test_config.py b/truss/tests/test_config.py index 2176f896b..d0f03ec91 100644 --- a/truss/tests/test_config.py +++ b/truss/tests/test_config.py @@ -230,6 +230,10 @@ def test_validate_mem_spec(mem_spec, expected_valid, memory_in_bytes): "RTX_PRO_6000", AcceleratorSpec(accelerator=Accelerator.RTX_PRO_6000, count=1), ), + ( + "RTX_PRO_6000_48GB_GFX", + AcceleratorSpec(accelerator=Accelerator.RTX_PRO_6000_48GB_GFX, count=1), + ), ("B300", AcceleratorSpec(accelerator=Accelerator.B300, count=1)), ("GB300:4", AcceleratorSpec(accelerator=Accelerator.GB300, count=4)), ], diff --git a/uv.lock b/uv.lock index 460d18ca0..8e558cd43 100644 --- a/uv.lock +++ b/uv.lock @@ -1,5 +1,5 @@ version = 1 -revision = 2 +revision = 3 requires-python = ">=3.9, <3.15" resolution-markers = [ "python_full_version >= '3.13'", @@ -3632,7 +3632,7 @@ wheels = [ [[package]] name = "truss" -version = "0.18.23" +version = "0.18.25rc0" source = { editable = "." } dependencies = [ { name = "aiofiles" },