Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(taesd): remove unused variable warning

Browse Source
This commit is contained in:
CalamitousFelicitousness
2026-01-16 03:06:48 +00:00
parent 749371364b
commit 5e2bc01367
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_vae_taesd.py
+1 -1
View File
@@ -161,7 +161,7 @@ def decode(latents):
shared.log.debug(f'Decode: type="taesd" variant="{variant}" input={latents.shape} tensor={tensor.shape}')
# FLUX.2 has 128 latent channels that need reshaping to 32 channels for TAESD
if variant == 'TAE FLUX.2' and len(tensor.shape) == 4 and tensor.shape[1] == 128:
b, c, h, w = tensor.shape
b, _c, h, w = tensor.shape
tensor = tensor.reshape(b, 32, h * 2, w * 2)
if variant.startswith('TAESD') or variant in {'TAE FLUX.1', 'TAE FLUX.2', 'TAE SD3'}:
image = vae.decoder(tensor).clamp(0, 1).detach()