[Diffusion] Remove useless dependency in diffusion (#16967)
This commit is contained in:
@@ -93,7 +93,6 @@ diffusion = [
|
||||
"remote-pdb",
|
||||
"st_attn ==0.0.7",
|
||||
"vsa==0.0.4",
|
||||
"yunchang==0.6.3.post1",
|
||||
"runai_model_streamer",
|
||||
"cache-dit==1.1.8"
|
||||
]
|
||||
|
||||
@@ -418,7 +418,7 @@ def initialize_model_parallel(
|
||||
|
||||
PROCESS_GROUP = _DummyProcessGroup()
|
||||
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
|
||||
# (e.g., tp-sp order).
|
||||
sp_groups = rank_generator.get_ranks("sp")
|
||||
|
||||
Reference in New Issue
Block a user