diff --git a/requirements/cuda.txt b/requirements/cuda.txt index 2a8b2734f8..1776723560 100644 --- a/requirements/cuda.txt +++ b/requirements/cuda.txt @@ -22,7 +22,7 @@ nvidia-cudnn-frontend>=1.19.1 # Required for LLM_NVTX_SCOPES_FOR_PROFILING=1 nvtx==0.2.15 # Required for faster safetensors model loading -fastsafetensors >= 0.3.2 +fastsafetensors >= 0.3.3 # QuACK and Cutlass DSL for FA4 (cute-DSL implementation) nvidia-cutlass-dsl[cu13]==4.6.0 diff --git a/requirements/rocm.txt b/requirements/rocm.txt index b5bcdf4b33..d2bf9898a8 100644 --- a/requirements/rocm.txt +++ b/requirements/rocm.txt @@ -26,4 +26,4 @@ tilelang==0.1.10 # Required apache-tvm-ffi matching tilelang version apache-tvm-ffi==0.1.10 # Required for faster safetensors model loading -fastsafetensors >= 0.3.2 +fastsafetensors >= 0.3.3 diff --git a/requirements/test/cpu.txt b/requirements/test/cpu.txt index 0237ba165b..08e7237951 100644 --- a/requirements/test/cpu.txt +++ b/requirements/test/cpu.txt @@ -234,7 +234,7 @@ fastparquet==2024.11.0 # via genai-perf fastrlock==0.8.2 # via cupy-cuda12x -fastsafetensors==0.3.2 +fastsafetensors==0.3.3 # via -r requirements/test/cuda.in filelock==3.16.1 # via diff --git a/requirements/test/cuda.in b/requirements/test/cuda.in index 1a652a035f..116f405444 100644 --- a/requirements/test/cuda.in +++ b/requirements/test/cuda.in @@ -57,7 +57,7 @@ grpcio-reflection==1.78.0 arctic-inference == 0.1.1; platform_machine == "x86_64" # Required for suffix decoding test numba == 0.65.0 # Required for N-gram speculative decoding runai-model-streamer[s3,gcs,azure]==0.15.7 -fastsafetensors>=0.3.2 +fastsafetensors>=0.3.3 instanttensor>=0.1.9; platform_machine == "x86_64" decord==0.6.0; platform_machine == "x86_64" # terratorch is temporarily disabled while PyPI has the `lightning` package diff --git a/requirements/test/cuda.txt b/requirements/test/cuda.txt index 4952fba64d..d722581275 100644 --- a/requirements/test/cuda.txt +++ b/requirements/test/cuda.txt @@ -250,7 +250,7 @@ fastparquet==2024.11.0 # via genai-perf fastrlock==0.8.2 # via cupy-cuda12x -fastsafetensors==0.3.2 +fastsafetensors==0.3.3 # via # -c requirements/cuda.txt # -r requirements/test/cuda.in diff --git a/requirements/test/nightly-torch.txt b/requirements/test/nightly-torch.txt index 1beb4294d0..a98dca7394 100644 --- a/requirements/test/nightly-torch.txt +++ b/requirements/test/nightly-torch.txt @@ -43,6 +43,6 @@ tritonclient>=2.51.0 numba == 0.65.0 # Required for N-gram speculative decoding numpy runai-model-streamer[s3,gcs,azure]==0.15.7 -fastsafetensors>=0.3.2 +fastsafetensors>=0.3.3 instanttensor>=0.1.9 pydantic>=2.12 # 2.11 leads to error on python 3.13 diff --git a/requirements/test/rocm.in b/requirements/test/rocm.in index 9edb08dc46..f8a0ece15b 100644 --- a/requirements/test/rocm.in +++ b/requirements/test/rocm.in @@ -53,7 +53,7 @@ grpcio-reflection==1.78.0 arctic-inference==0.1.1 # Required for suffix decoding test numba==0.65.0 # Required for N-gram speculative decoding runai-model-streamer[s3,gcs,azure]==0.15.7 -fastsafetensors>=0.3.2 +fastsafetensors>=0.3.3 instanttensor>=0.1.9 decord==0.6.0 diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index 43283f8ff6..9b00aee54d 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -237,7 +237,7 @@ fastavro==1.12.2 # via cohere fastparquet==2026.3.0 # via genai-perf -fastsafetensors==0.3.2 +fastsafetensors==0.3.3 # via # -c requirements/rocm.txt # -r requirements/test/rocm.in diff --git a/setup.py b/setup.py index 587f66dfbe..b88a4e8b52 100644 --- a/setup.py +++ b/setup.py @@ -1272,7 +1272,7 @@ setup( "zen": ["zentorch==2.11.0.0"], "bench": ["pandas", "matplotlib", "seaborn", "datasets", "scipy", "plotly"], "tensorizer": ["tensorizer==2.10.1"], - "fastsafetensors": ["fastsafetensors >= 0.3.2"], + "fastsafetensors": ["fastsafetensors >= 0.3.3"], "instanttensor": ["instanttensor >= 0.1.9"], "runai": ["runai-model-streamer[s3,gcs,azure] >= 0.15.7"], "audio": [