From 7553b7dcb01e57fc96a1fd1378d5b75f05c32437 Mon Sep 17 00:00:00 2001 From: R0CKSTAR Date: Thu, 19 Mar 2026 10:39:16 +0800 Subject: [PATCH] chore: extract diffusion_common in python/pyproject_other.toml (#20803) Signed-off-by: Xiaodong Ye --- python/pyproject_other.toml | 48 ++++++++++++++----------------------- 1 file changed, 18 insertions(+), 30 deletions(-) diff --git a/python/pyproject_other.toml b/python/pyproject_other.toml index cbf20082e..60b5a0089 100755 --- a/python/pyproject_other.toml +++ b/python/pyproject_other.toml @@ -65,6 +65,21 @@ runtime_common = [ "smg-grpc-servicer>=0.5.0", ] +diffusion_common = [ + "PyYAML==6.0.1", + "cloudpickle", + "diffusers==0.37.0", + "imageio==2.36.0", + "imageio-ffmpeg==0.5.1", + "moviepy>=2.0.0", + "opencv-python-headless==4.10.0.84", + "remote-pdb", + "addict", + "scikit-image==0.25.2", + "trimesh>=4.0.0", + "xatlas", +] + tracing = [ "opentelemetry-sdk", "opentelemetry-api", @@ -82,22 +97,11 @@ srt_hip = [ ] diffusion_hip = [ - "PyYAML==6.0.1", - "cloudpickle", - "diffusers==0.37.0", - "imageio==2.36.0", - "imageio-ffmpeg==0.5.1", - "moviepy>=2.0.0", - "opencv-python-headless==4.10.0.84", - "remote-pdb", + "sglang[diffusion_common]", "st_attn==0.0.7", "vsa==0.0.4", "runai_model_streamer>=0.15.5", "cache-dit==1.1.8", - "addict", - "scikit-image==0.25.2", - "trimesh>=4.0.0", - "xatlas", ] # For Intel Gaudi(device : hpu) follow the installation guide @@ -115,22 +119,11 @@ srt_musa = [ ] diffusion_musa = [ - "PyYAML==6.0.1", - "cloudpickle", - "diffusers==0.37.0", - "imageio==2.36.0", - "imageio-ffmpeg==0.5.1", - "moviepy>=2.0.0", - "opencv-python-headless==4.10.0.84", - "remote-pdb", + "sglang[diffusion_common]", "st_attn==0.0.7", "vsa==0.0.4", "runai_model_streamer>=0.15.5", "cache-dit==1.1.8", - "addict", - "scikit-image==0.25.2", - "trimesh>=4.0.0", - "xatlas", ] # https://docs.sglang.io/platforms/mps.md @@ -143,13 +136,8 @@ srt_mps = [ ] diffusion_mps = [ - "PyYAML==6.0.1", + "sglang[diffusion_common]", "cloudpickle==3.1.2", - "diffusers==0.37.0", - "imageio==2.36.0", - "imageio-ffmpeg==0.5.1", - "moviepy>=2.0.0", - "opencv-python-headless==4.10.0.84", "remote-pdb==2.1.0", "cache-dit==1.2.3", "addict==2.4.0",