minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion] fix: fix comfyui import typo (#17834)

Browse Source
This commit is contained in:
triple-mu
2026-01-28 23:49:55 +08:00
committed by GitHub
parent f8636fbb25
commit 1d1e72e516
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/runtime/pipelines/comfyui_qwen_image_pipeline.py
+5 -3
View File
@@ -12,10 +12,12 @@ from sglang.multimodal_gen.configs.models.dits.qwenimage import QwenImageDitConf
from sglang.multimodal_gen.runtime.distributed import get_local_torch_device from sglang.multimodal_gen.runtime.distributed import get_local_torch_device
from sglang.multimodal_gen.runtime.loader.fsdp_load import ( from sglang.multimodal_gen.runtime.loader.fsdp_load import (
load_model_from_full_model_state_dict, load_model_from_full_model_state_dict,
set_default_dtype,
shard_model, shard_model,
) )
from sglang.multimodal_gen.runtime.loader.utils import get_param_names_mapping from sglang.multimodal_gen.runtime.loader.utils import (
get_param_names_mapping,
set_default_torch_dtype,
)
from sglang.multimodal_gen.runtime.loader.weight_utils import ( from sglang.multimodal_gen.runtime.loader.weight_utils import (
safetensors_weights_iterator, safetensors_weights_iterator,
) )
@@ -222,7 +224,7 @@ class ComfyUIQwenImagePipelineBase(LoRAPipeline, ComposedPipelineBase):
mp_policy=mp_policy, mp_policy=mp_policy,
) )
with set_default_dtype(default_dtype), torch.device("meta"): with set_default_torch_dtype(default_dtype), torch.device("meta"):
model = model_cls(**{"config": dit_config, "hf_config": hf_config}) model = model_cls(**{"config": dit_config, "hf_config": hf_config})
use_fsdp = server_args.use_fsdp_inference use_fsdp = server_args.use_fsdp_inference