From 1d1e72e516f7f9df2f682283ee9e35692e2b9e66 Mon Sep 17 00:00:00 2001 From: triple-mu Date: Thu, 29 Jan 2026 00:49:55 +0900 Subject: [PATCH] [diffusion] fix: fix comfyui import typo (#17834) --- .../runtime/pipelines/comfyui_qwen_image_pipeline.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/python/sglang/multimodal_gen/runtime/pipelines/comfyui_qwen_image_pipeline.py b/python/sglang/multimodal_gen/runtime/pipelines/comfyui_qwen_image_pipeline.py index 071a8d197..0e2c031f7 100644 --- a/python/sglang/multimodal_gen/runtime/pipelines/comfyui_qwen_image_pipeline.py +++ b/python/sglang/multimodal_gen/runtime/pipelines/comfyui_qwen_image_pipeline.py @@ -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.loader.fsdp_load import ( load_model_from_full_model_state_dict, - set_default_dtype, 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 ( safetensors_weights_iterator, ) @@ -222,7 +224,7 @@ class ComfyUIQwenImagePipelineBase(LoRAPipeline, ComposedPipelineBase): 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}) use_fsdp = server_args.use_fsdp_inference