1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(glm): rename _wrap_vision_language_generate to hijack_vision_language_generate

pull/4548/head
CalamitousFelicitousness 2026-01-14 11:34:25 +00:00 committed by GitHub
parent de3b2b62eb
commit 7fb58f22c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pipelines/model_glm.py
View File

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