From 71b081ff19e7dad24b87658a84bc44ef0562d8a3 Mon Sep 17 00:00:00 2001 From: mgoin Date: Tue, 14 Apr 2026 17:11:57 +0000 Subject: [PATCH] Use instanttensor == 0.1.8 Signed-off-by: mgoin --- requirements/test/cuda.in | 2 +- requirements/test/cuda.txt | 2 +- requirements/test/nightly-torch.txt | 2 +- requirements/test/rocm.in | 2 +- requirements/test/rocm.txt | 2 +- setup.py | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) diff --git a/requirements/test/cuda.in b/requirements/test/cuda.in index 378ecf94222..4b255d41ea7 100644 --- a/requirements/test/cuda.in +++ b/requirements/test/cuda.in @@ -59,7 +59,7 @@ numba == 0.61.2 # Required for N-gram speculative decoding numpy runai-model-streamer[s3,gcs,azure]==0.15.7 fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage -instanttensor>=0.1.5 +instanttensor==0.1.8 pydantic>=2.12 # 2.11 leads to error on python 3.13 decord==0.6.0; platform_machine == "x86_64" terratorch >= 1.2.2 # Required for Prithvi tests diff --git a/requirements/test/cuda.txt b/requirements/test/cuda.txt index 548ca9310ff..0a69eb4600c 100644 --- a/requirements/test/cuda.txt +++ b/requirements/test/cuda.txt @@ -401,7 +401,7 @@ inflect==5.6.2 # via datamodel-code-generator iniconfig==2.0.0 # via pytest -instanttensor==0.1.5 +instanttensor==0.1.8 # via -r requirements/test/cuda.in isodate==0.7.2 # via azure-storage-blob diff --git a/requirements/test/nightly-torch.txt b/requirements/test/nightly-torch.txt index e0eb7e11411..724508afa1c 100644 --- a/requirements/test/nightly-torch.txt +++ b/requirements/test/nightly-torch.txt @@ -44,5 +44,5 @@ numba == 0.61.2 # Required for N-gram speculative decoding numpy runai-model-streamer[s3,gcs,azure]==0.15.7 fastsafetensors>=0.2.2 -instanttensor>=0.1.5 +instanttensor==0.1.8 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 b5a9451b36f..2f12d53b189 100644 --- a/requirements/test/rocm.in +++ b/requirements/test/rocm.in @@ -57,7 +57,7 @@ numba==0.61.2 # Required for N-gram speculative decoding numpy runai-model-streamer[s3,gcs,azure]==0.15.7 fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage -instanttensor>=0.1.5 +instanttensor==0.1.8 pydantic>=2.12 # 2.11 leads to error on python 3.13 decord==0.6.0 diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index e1efae912ee..1f5e8c84f54 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -467,7 +467,7 @@ inflect==7.5.0 # via datamodel-code-generator iniconfig==2.3.0 # via pytest -instanttensor==0.1.6 +instanttensor==0.1.8 # via -r requirements/test/rocm.in interegular==0.3.3 # via lm-format-enforcer diff --git a/setup.py b/setup.py index 1748781985f..088301a1984 100644 --- a/setup.py +++ b/setup.py @@ -1079,7 +1079,7 @@ setup( "bench": ["pandas", "matplotlib", "seaborn", "datasets", "scipy", "plotly"], "tensorizer": ["tensorizer==2.10.1"], "fastsafetensors": ["fastsafetensors >= 0.2.2"], - "instanttensor": ["instanttensor >= 0.1.5"], + "instanttensor": ["instanttensor == 0.1.8"], "runai": ["runai-model-streamer[s3,gcs,azure] >= 0.15.7"], "audio": [ "av",