Update setup.py

lucidrains · Jul 4, 2024 · a8880ad · a8880ad
1 parent c3f7c9f
commit a8880ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -22,7 +22,6 @@
   install_requires=[
     'matplotlib',
     'accelerate>=0.25.0',
-    'huggingface_hub>=0.21.4',
     'beartype',
     "huggingface_hub>=0.21.4",
     'classifier-free-guidance-pytorch>=0.6.10',
@@ -35,6 +34,7 @@
     'local-attention>=1.9.0',
     'numpy',
     'pytorch-custom-utils>=0.0.9',
+    'rotary-embedding-torch>=0.6.4',
     'taylor-series-linear-attention>=0.1.6',
     'torch>=2.1',
     'torch_geometric',