diff --git a/requirements.txt b/requirements.txt index 2abba8b..85ecc8c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ gluonts>=0.10.0 pytorch-lightning datasets xformers -https://github.com/ml-jku/hopfield-layers +git+https://github.com/ml-jku/hopfield-layers etsformer-pytorch reformer_pytorch einops @@ -12,4 +12,4 @@ opt_einsum pykeops scipy apex -torchscale +git+https://github.com/microsoft/torchscale