diff --git a/requirements.txt b/requirements.txt index 4c2c0b2b2..3bc945a1b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,6 +2,7 @@ torch torchsde torchvision torchaudio +numpy>=1.25.0 einops transformers>=4.28.1 tokenizers>=0.13.3