add GLM-Image pipeline support

- Add GLM-Image (zai-org/GLM-Image) model detection and loading
- Custom pipeline loader with proper component handling:
  - ByT5 text encoder (cannot use shared T5 due to different hidden size)
  - Vision-language encoder (9B AR model)
  - DiT transformer (7B)
- Fix EOS token early stopping in AR generation
- Add AR token generation progress tracking with terminal progress bar
- Fix uninitialized audio variable in processing
- Add TAESD support for GLM-Image (using f1 variant)
This commit is contained in:
CalamitousFelicitousness
2026-01-14 03:33:49 +00:00
parent 8500156888
commit 3f259cff9a
8 changed files with 151 additions and 2 deletions
+2
View File
@@ -78,6 +78,8 @@ def get_model_type(pipe):
model_type = 'prx'
elif 'LongCat' in name:
model_type = 'longcat'
elif 'GlmImage' in name:
model_type = 'glm_image'
elif 'Ovis-Image' in name:
model_type = 'ovis'
# video models