mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-30 04:12:37 +08:00
Fix subclassing the wrong thing for the Flux2 TAE encoder
This commit is contained in:
parent
c831b0f5fa
commit
f49398110a
@ -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")
|
||||
|
||||
Loading…
Reference in New Issue
Block a user