Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(ideogram4): offload-safe encode, step progress, live preview

Browse Source 
- move the text encoder on-device for the tapped forward (bypasses the offload hook)
- wrap the denoise loop in the diffusers progress bar
- live preview through the shared TAE FLUX.2 decoder (Flux.2 VAE)
This commit is contained in:
CalamitousFelicitousness
2026-06-04 02:36:42 +01:00
parent b5fdb1995e
commit 4866d2c8f3
2 changed files with 46 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/vae/sd_vae_taesd.py
+1 -1
View File
@@ -71,7 +71,7 @@ def get_model(model_cls, variant=None):
elif model_cls in {'f1', 'h1', 'zimage', 'lumina2', 'chroma', 'longcat', 'omnigen2', 'flite', 'ovis', 'kandinsky5', 'glmimage', 'cogview3', 'cogview4', 'ultraflux'}:
model_cls = 'f1'
variant = 'TAE FLUX.1'
elif model_cls in {'f2', 'ernieimage', 'lens'}:
elif model_cls in {'f2', 'ernieimage', 'lens', 'ideogram4'}:
model_cls = 'f2'
variant = 'TAE FLUX.2'
elif model_cls in {'sd3'}: