chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

[Diffusion] switch to local calculate_dimensions (#13294)

Browse Source
This commit is contained in:
Adarsh Shirawalmath
2025-11-15 13:55:13 +08:00
committed by GitHub
parent 10285ec204
commit c2083116e3
3 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/configs/models/vaes/qwenimage.py
+1 -2
View File
@@ -3,9 +3,8 @@
# SPDX-License-Identifier: Apache-2.0
from dataclasses import dataclass, field
from diffusers.pipelines.qwenimage.pipeline_qwenimage_edit import calculate_dimensions
from sglang.multimodal_gen.configs.models.vaes.base import VAEArchConfig, VAEConfig
from sglang.multimodal_gen.utils import calculate_dimensions
@dataclass
python/sglang/multimodal_gen/configs/pipelines/qwen_image.py
+1 -1
View File
@@ -4,13 +4,13 @@ from dataclasses import dataclass, field
from typing import Callable
import torch
from diffusers.pipelines.qwenimage.pipeline_qwenimage_edit import calculate_dimensions
from sglang.multimodal_gen.configs.models import DiTConfig, EncoderConfig, VAEConfig
from sglang.multimodal_gen.configs.models.dits.qwenimage import QwenImageDitConfig
from sglang.multimodal_gen.configs.models.encoders.qwen_image import Qwen2_5VLConfig
from sglang.multimodal_gen.configs.models.vaes.qwenimage import QwenImageVAEConfig
from sglang.multimodal_gen.configs.pipelines.base import ModelTaskType, PipelineConfig
from sglang.multimodal_gen.utils import calculate_dimensions
def _extract_masked_hidden(hidden_states: torch.Tensor, mask: torch.Tensor):