From 7fb58f22c22465d225f0697698d534426e2c592c Mon Sep 17 00:00:00 2001 From: CalamitousFelicitousness Date: Wed, 14 Jan 2026 11:34:25 +0000 Subject: [PATCH] fix(glm): rename _wrap_vision_language_generate to hijack_vision_language_generate --- pipelines/model_glm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pipelines/model_glm.py b/pipelines/model_glm.py index d9b45ceb1..31974f75b 100644 --- a/pipelines/model_glm.py +++ b/pipelines/model_glm.py @@ -54,7 +54,7 @@ class GLMTokenProgressProcessor(transformers.LogitsProcessor): return scores -def _wrap_vision_language_generate(pipe): +def hijack_vision_language_generate(pipe): """Wrap vision_language_encoder.generate to add progress tracking.""" if not hasattr(pipe, 'vision_language_encoder') or pipe.vision_language_encoder is None: return @@ -132,6 +132,6 @@ def load_glm_image(checkpoint_info, diffusers_load_config=None): del transformer, text_encoder, vision_language_encoder sd_hijack_te.init_hijack(pipe) - _wrap_vision_language_generate(pipe) # Add progress tracking for AR token generation + hijack_vision_language_generate(pipe) # Add progress tracking for AR token generation devices.torch_gc(force=True, reason='load') return pipe