[diffusion]: align sglang diffusion AMD pyproject_other.toml diffusion dependency with pyproject.toml (#16225)

Co-authored-by: roywang <roywang@amd.com>
This commit is contained in:
RoyWang
2026-01-29 17:50:57 +08:00
committed by GitHub
parent ef1c512754
commit 30adf78f82
4 changed files with 8 additions and 13 deletions

View File

@@ -85,18 +85,18 @@ srt_hip = [
]
diffusion_hip = [
"diffusers @ git+https://github.com/huggingface/diffusers.git@6290fdfda40610ce7b99920146853614ba529c6e",
"opencv-python-headless==4.10.0.84",
"PyYAML==6.0.1",
"cloudpickle",
"diffusers==0.36.0",
"imageio==2.36.0",
"imageio-ffmpeg==0.5.1",
"PyYAML==6.0.1",
"moviepy>=2.0.0",
"cloudpickle",
"opencv-python-headless==4.10.0.84",
"remote-pdb",
"torchcodec==0.5.0",
"st_attn==0.0.7",
"vsa==0.0.4",
"runai_model_streamer>=0.15.5",
"cache-dit==1.1.8"
]
# For Intel Gaudi(device : hpu) follow the installation guide
@@ -141,7 +141,7 @@ test = [
"tabulate",
]
all_hip = ["sglang[srt_hip]"]
all_hip = ["sglang[srt_hip]", "sglang[diffusion_hip]"]
all_hpu = ["sglang[srt_hpu]"]
all_musa = ["sglang[srt_musa]", "sglang[diffusion_musa]"]

View File

@@ -23,11 +23,6 @@ try:
except ImportError:
HAS_RUNAI_MODEL_STREAMER = False
# Disable runai_model_streamer on AMD/ROCm due to global state issues
# that cause "Streamer is handling previous request" errors
if torch.version.hip is not None:
HAS_RUNAI_MODEL_STREAMER = False
from sglang.multimodal_gen.runtime.distributed import get_local_torch_device
from sglang.multimodal_gen.runtime.utils.logging_utils import init_logger