AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[chore] make transformers version check stricter for glm image. (#12974)

Browse Source 
* make transformers version check stricter for glm image.

* public checkpoint.
This commit is contained in:
Sayak Paul
2026-01-14 10:29:48 +05:30
committed by GitHub
parent 3c70440d26
commit 7feb4fc791
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/glm_image/pipeline_glm_image.py
View File

@@ -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

5
tests/pipelines/glm_image/test_glm_image.py
View File

@@ -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: