wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-30 20:32:45 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Fix subclassing the wrong thing for the Flux2 TAE encoder

Browse Source
This commit is contained in:
blepping 2026-04-21 07:02:39 -06:00
parent c831b0f5fa
commit f49398110a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/taesd/taesd.py
View File

@ -74,7 +74,7 @@ class DecoderFlux2(Decoder):
) )
return super().forward(x) return super().forward(x)
class EncoderFlux2(Decoder): class EncoderFlux2(Encoder):
def __init__(self, latent_channels: int = 128, use_gn: bool = True): def __init__(self, latent_channels: int = 128, use_gn: bool = True):
if latent_channels != 128 or not use_gn: if latent_channels != 128 or not use_gn:
raise ValueError("Unexpected parameters for Flux2 TAE module") raise ValueError("Unexpected parameters for Flux2 TAE module")