update setup.py from uriels PR #668

facebookresearch · Mar 12, 2023 · cdec8fa · cdec8fa
1 parent 5b2de08
commit cdec8fa
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -189,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,