diff --git a/requirements.txt b/requirements.txt index d2187bcc..8e6f232c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,7 +6,7 @@ pydantic torch>=2.1.0 # xformers>=0.0.19 torchvision -torchaudio +# torchaudio torchdiffeq torchsde einops