[diffusion][llm] macOS support (#19549)
Signed-off-by: Xiaodong Ye <yeahdongcn@gmail.com> Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
@@ -136,6 +136,29 @@ diffusion_musa = [
|
||||
"xatlas",
|
||||
]
|
||||
|
||||
# https://docs.sglang.io/platforms/mps.md
|
||||
srt_mps = [
|
||||
"sglang[runtime_common]",
|
||||
"torch==2.9.1",
|
||||
"torchao==0.9.0",
|
||||
"torchaudio==2.9.1",
|
||||
"torchvision",
|
||||
]
|
||||
|
||||
diffusion_mps = [
|
||||
"PyYAML==6.0.1",
|
||||
"cloudpickle==3.1.2",
|
||||
"diffusers==0.36.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",
|
||||
"av==16.1.0",
|
||||
]
|
||||
|
||||
test = [
|
||||
"accelerate",
|
||||
"expecttest",
|
||||
@@ -152,10 +175,12 @@ test = [
|
||||
all_hip = ["sglang[srt_hip]", "sglang[diffusion_hip]"]
|
||||
all_hpu = ["sglang[srt_hpu]"]
|
||||
all_musa = ["sglang[srt_musa]", "sglang[diffusion_musa]"]
|
||||
all_mps = ["sglang[srt_mps]", "sglang[diffusion_mps]"]
|
||||
|
||||
dev_hip = ["sglang[all_hip]", "sglang[test]"]
|
||||
dev_hpu = ["sglang[all_hpu]", "sglang[test]"]
|
||||
dev_musa = ["sglang[all_musa]", "sglang[test]"]
|
||||
dev_mps = ["sglang[all_mps]", "sglang[test]"]
|
||||
|
||||
[project.urls]
|
||||
"Homepage" = "https://github.com/sgl-project/sglang"
|
||||
|
||||
Reference in New Issue
Block a user