Skip to content

Commit 7feb4fc

Browse files
authored
[chore] make transformers version check stricter for glm image. (#12974)
* make transformers version check stricter for glm image. * public checkpoint.
1 parent 3c70440 commit 7feb4fc

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

src/diffusers/pipelines/glm_image/pipeline_glm_image.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
# Because it's not released in stable as of 13/01/2026. So this is just a proxy.
3737
GlmImageProcessor = ProcessorMixin
3838
GlmImageForConditionalGeneration = PreTrainedModel
39-
if is_transformers_version(">=", "4.57.4"):
39+
if is_transformers_version(">=", "5.0.0.dev0"):
4040
from transformers import GlmImageForConditionalGeneration, GlmImageProcessor
4141

4242

tests/pipelines/glm_image/test_glm_image.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
from ..test_pipelines_common import PipelineTesterMixin
2727

2828

29-
if is_transformers_version(">=", "4.57.4"):
29+
if is_transformers_version(">=", "5.0.0.dev0"):
3030
from transformers import GlmImageConfig, GlmImageForConditionalGeneration, GlmImageProcessor
3131

3232

@@ -86,8 +86,7 @@ def get_dummy_components(self):
8686
torch.manual_seed(0)
8787
vision_language_encoder = GlmImageForConditionalGeneration(glm_config)
8888

89-
# TODO: move to a public checkpoint
90-
processor = GlmImageProcessor.from_pretrained("ZP2Test/GLM-Image", subfolder="processor")
89+
processor = GlmImageProcessor.from_pretrained("zai-org/GLM-Image", subfolder="processor")
9190

9291
torch.manual_seed(0)
9392
# For GLM-Image, the relationship between components must satisfy:

0 commit comments

Comments
 (0)