Skip to content

Commit 6a1904e

Browse files
[bug fix] GLM-Image fit new get_image_features API (#13052)
change get_image_features API Signed-off-by: JaredforReal <w13431838023@gmail.com> Co-authored-by: YiYi Xu <yixu310@gmail.com>
1 parent f5b6b66 commit 6a1904e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/glm_image/pipeline_glm_image.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -407,8 +407,8 @@ def generate_prior_tokens(
407407

408408
if len(source_grids) > 0:
409409
prior_token_image_embed = self.vision_language_encoder.get_image_features(
410-
inputs["pixel_values"], source_grids, return_dict=False
411-
)
410+
inputs["pixel_values"], source_grids
411+
).pooler_output
412412
prior_token_image_embed = torch.cat(prior_token_image_embed, dim=0)
413413
prior_token_image_ids_d32 = self.vision_language_encoder.get_image_tokens(
414414
prior_token_image_embed, source_grids

0 commit comments

Comments
 (0)