From d34b411313c6b13da08d4e80fa92e29c7310faba Mon Sep 17 00:00:00 2001 From: JaredforReal Date: Thu, 29 Jan 2026 10:10:45 +0800 Subject: [PATCH] change get_image_features API Signed-off-by: JaredforReal --- src/diffusers/pipelines/glm_image/pipeline_glm_image.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/glm_image/pipeline_glm_image.py b/src/diffusers/pipelines/glm_image/pipeline_glm_image.py index 8cbaf99a2243..589b3be47b2c 100644 --- a/src/diffusers/pipelines/glm_image/pipeline_glm_image.py +++ b/src/diffusers/pipelines/glm_image/pipeline_glm_image.py @@ -407,8 +407,8 @@ def generate_prior_tokens( if len(source_grids) > 0: prior_token_image_embed = self.vision_language_encoder.get_image_features( - inputs["pixel_values"], source_grids, return_dict=False - ) + inputs["pixel_values"], source_grids + ).pooler_output prior_token_image_embed = torch.cat(prior_token_image_embed, dim=0) prior_token_image_ids_d32 = self.vision_language_encoder.get_image_tokens( prior_token_image_embed, source_grids