update setup.py

Author: suchenzang

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,