update setup.py

facebookresearch · Mar 5, 2023 · 2814075 · 2814075
1 parent c49c4a0
commit 2814075
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/setup.py b/setup.py
@@ -169,13 +169,6 @@ def do_setup(package_data):
             "torch",
             "tqdm",
             "typing_extensions",
-            "einops",
-            "webdataset==0.1.103",
-            "matplotlib",
-            "pytorchvideo==0.1.5",
-            "wandb",
-            "albumentations",
-            "dalle_pytorch",
         ],
         dependency_links=dependency_links,
         packages=find_packages(
@@ -196,6 +189,15 @@ def do_setup(package_data):
                 "pyarrow",
                 "boto3",
                 "pandas",
+            ],
+            "multimodal": [
+                "albumentations",
+                "dalle_pytorch",
+                "einops",
+                "matplotlib",
+                "pytorchvideo==0.1.5",
+                "wandb",
+                "webdataset==0.1.103",
             ]
         },
         package_data=package_data,