From 7feb4fc791909f4a58b153d6d07c6cb8437b2e58 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Wed, 14 Jan 2026 10:29:48 +0530 Subject: [PATCH] [chore] make transformers version check stricter for glm image. (#12974) * make transformers version check stricter for glm image. * public checkpoint. --- src/diffusers/pipelines/glm_image/pipeline_glm_image.py | 2 +- tests/pipelines/glm_image/test_glm_image.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/src/diffusers/pipelines/glm_image/pipeline_glm_image.py b/src/diffusers/pipelines/glm_image/pipeline_glm_image.py index 57fb84c677..5499b8769f 100644 --- a/src/diffusers/pipelines/glm_image/pipeline_glm_image.py +++ b/src/diffusers/pipelines/glm_image/pipeline_glm_image.py @@ -36,7 +36,7 @@ from .pipeline_output import GlmImagePipelineOutput # Because it's not released in stable as of 13/01/2026. So this is just a proxy. GlmImageProcessor = ProcessorMixin GlmImageForConditionalGeneration = PreTrainedModel -if is_transformers_version(">=", "4.57.4"): +if is_transformers_version(">=", "5.0.0.dev0"): from transformers import GlmImageForConditionalGeneration, GlmImageProcessor diff --git a/tests/pipelines/glm_image/test_glm_image.py b/tests/pipelines/glm_image/test_glm_image.py index cebb71d602..7a380b99b0 100644 --- a/tests/pipelines/glm_image/test_glm_image.py +++ b/tests/pipelines/glm_image/test_glm_image.py @@ -26,7 +26,7 @@ from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_IMAGE_PA from ..test_pipelines_common import PipelineTesterMixin -if is_transformers_version(">=", "4.57.4"): +if is_transformers_version(">=", "5.0.0.dev0"): from transformers import GlmImageConfig, GlmImageForConditionalGeneration, GlmImageProcessor @@ -86,8 +86,7 @@ class GlmImagePipelineFastTests(PipelineTesterMixin, unittest.TestCase): torch.manual_seed(0) vision_language_encoder = GlmImageForConditionalGeneration(glm_config) - # TODO: move to a public checkpoint - processor = GlmImageProcessor.from_pretrained("ZP2Test/GLM-Image", subfolder="processor") + processor = GlmImageProcessor.from_pretrained("zai-org/GLM-Image", subfolder="processor") torch.manual_seed(0) # For GLM-Image, the relationship between components must satisfy: