diff --git a/requirements.txt b/requirements.txt index 7686a5f8a..e51ab2492 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