update setup.py

facebookresearch · Mar 5, 2023 · a4afe77 · a4afe77
1 parent c49c4a0
commit a4afe77
Showing 1 changed file with 10 additions and 8 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,7 +189,16 @@ 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,
         ext_modules=extensions,