Skip to content

Commit def3956

Browse files
tzhouamroot
andauthored
[Debug] Update GLM-Image Pipeline (#1049)
Co-authored-by: root <[email protected]>
1 parent cac6504 commit def3956

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

vllm_omni/diffusion/models/glm_image/pipeline_glm_image.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -491,7 +491,7 @@ def generate_prior_tokens(
491491
condition_grid = image_grid_thw[:-1]
492492
prior_token_image_embed = self.vision_language_encoder.get_image_features(
493493
inputs["pixel_values"], condition_grid
494-
)
494+
).pooler_output
495495
prior_token_image_embed = torch.cat(prior_token_image_embed, dim=0)
496496
flat_prior_token_image_ids = self.vision_language_encoder.get_image_tokens(
497497
prior_token_image_embed, condition_grid
@@ -859,7 +859,7 @@ def forward(self, req: OmniDiffusionRequest) -> DiffusionOutput:
859859
preprocessed_images = (
860860
None
861861
if isinstance(first_prompt, str)
862-
else first_prompt.get("additional_information", {}).get("preprocessed_image")
862+
else [first_prompt.get("additional_information", {}).get("preprocessed_image")]
863863
)
864864
condition_images = (
865865
None

0 commit comments

Comments
 (0)