From f49398110aee3beeda6f9db83ed409eebd27bca6 Mon Sep 17 00:00:00 2001 From: blepping Date: Tue, 21 Apr 2026 07:02:39 -0600 Subject: [PATCH] Fix subclassing the wrong thing for the Flux2 TAE encoder --- comfy/taesd/taesd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/taesd/taesd.py b/comfy/taesd/taesd.py index 80121a637..05d370209 100644 --- a/comfy/taesd/taesd.py +++ b/comfy/taesd/taesd.py @@ -74,7 +74,7 @@ class DecoderFlux2(Decoder): ) return super().forward(x) -class EncoderFlux2(Decoder): +class EncoderFlux2(Encoder): def __init__(self, latent_channels: int = 128, use_gn: bool = True): if latent_channels != 128 or not use_gn: raise ValueError("Unexpected parameters for Flux2 TAE module")