[Diffusion] Remove useless dependency in diffusion (#16967)
This commit is contained in:
@@ -93,7 +93,6 @@ diffusion = [
|
|||||||
"remote-pdb",
|
"remote-pdb",
|
||||||
"st_attn ==0.0.7",
|
"st_attn ==0.0.7",
|
||||||
"vsa==0.0.4",
|
"vsa==0.0.4",
|
||||||
"yunchang==0.6.3.post1",
|
|
||||||
"runai_model_streamer",
|
"runai_model_streamer",
|
||||||
"cache-dit==1.1.8"
|
"cache-dit==1.1.8"
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -418,7 +418,7 @@ def initialize_model_parallel(
|
|||||||
|
|
||||||
PROCESS_GROUP = _DummyProcessGroup()
|
PROCESS_GROUP = _DummyProcessGroup()
|
||||||
else:
|
else:
|
||||||
# Build yunchang SP sub-groups based on the true SP groups. This is
|
# Build SGLang Diffusion SP sub-groups based on the true SP groups. This is
|
||||||
# critical when TP>1, because SP groups may be strided in global ranks
|
# critical when TP>1, because SP groups may be strided in global ranks
|
||||||
# (e.g., tp-sp order).
|
# (e.g., tp-sp order).
|
||||||
sp_groups = rank_generator.get_ranks("sp")
|
sp_groups = rank_generator.get_ranks("sp")
|
||||||
|
|||||||
Reference in New Issue
Block a user