diff --git a/requirements.txt b/requirements.txt index 032371214..5b4738e7a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -41,10 +41,8 @@ safetensors scipy tb_nightly toml -torch torchdiffeq torchsde -torchvision voluptuous yapf scikit-image