fix dependencies

xdit-project · Feb 6, 2025 · dc988cd · dc988cd
1 parent 8378086
commit dc988cd
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -34,19 +34,25 @@ def get_cuda_version():
             "distvae",
             "yunchang>=0.6.0",
             "pytest",
-            "flask",
             "opencv-python",
             "imageio",
             "imageio-ffmpeg",
-            "optimum-quanto",
-            "ray"
         ],
         extras_require={
             "diffusers": [
                 "diffusers>=0.31.0",  # NOTE: diffusers>=0.32.0.dev is necessary for CogVideoX and Flux
             ],
             "flash-attn": [
-                "flash-attn>=2.6.0",
+                "flash-attn>=2.6.0",  # NOTE: flash-attn is necessary if ring_degree > 1
+            ],
+            "optimum-quanto": [
+                "optimum-quanto",  # NOTE: optimum-quanto is necessary if use_fp8_t5_encoder is enabled
+            ],
+            "flask": [
+                "flask",  # NOTE: flask is necessary to run xDiT as an http service
+            ],
+            "ray": [
+                "ray",  # NOTE: ray is necessary if RayDiffusionPipeline is used
             ]
         },
         url="https://github.com/xdit-project/xDiT.",