[Misc] Bump fastsafetensors version for latest fixes (#34273)

Signed-off-by: Nick Hill <nickhill123@gmail.com> (cherry picked from commit 79504027ef)
2026-02-11 00:30:09 -08:00
parent 9be1ff2d3a
commit 5e8adb0c49
4 changed files with 4 additions and 5 deletions
--- a/requirements/test.in
+++ b/requirements/test.in
@@ -53,7 +53,7 @@ arctic-inference == 0.1.1 # Required for suffix decoding test
 numba == 0.61.2 # Required for N-gram speculative decoding
 numpy
 runai-model-streamer[s3,gcs]==0.15.3
-fastsafetensors>=0.1.10
+fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage
 pydantic>=2.12 # 2.11 leads to error on python 3.13
 decord==0.6.0
 terratorch @ git+https://github.com/IBM/terratorch.git@1.1.rc3 # required for PrithviMAE test