diff --git a/requirements.txt b/requirements.txt index 49567ad61..97a3a00ca 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ torch torchsde torchvision torchaudio -numpy>=1.25.0 +numpy==2.2.6 einops transformers>=4.50.3 tokenizers>=0.13.3