Skip to content

Commit db57361

Browse files
committed
Compare against wan original decoder
1 parent 87817d0 commit db57361

4 files changed

Lines changed: 697 additions & 6 deletions

File tree

src/maxdiffusion/models/wan/autoencoder_kl_wan.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -967,7 +967,6 @@ def _count_conv3d(module):
967967
self._enc_conv_idx = [0]
968968
self._enc_feat_map = [None] * self._enc_conv_num
969969

970-
971970
class AutoencoderKLWan(nnx.Module, FlaxModelMixin, ConfigMixin):
972971

973972
def __init__(

0 commit comments

Comments
 (0)