We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 557ee7d commit 211f58fCopy full SHA for 211f58f
1 file changed
src/maxdiffusion/models/wan/autoencoder_kl_wan.py
@@ -177,7 +177,6 @@ class WanResample(nnx.Module):
177
def __init__(self, dim: int, mode: str, rngs: nnx.Rngs, mesh: jax.sharding.Mesh = None, dtype: jnp.dtype = jnp.float32, weights_dtype: jnp.dtype = jnp.float32, precision: jax.lax.Precision = None):
178
self.dim = dim
179
self.mode = mode
180
- self.time_conv = None
181
182
if mode == "upsample2d":
183
self.upsample = WanUpsample(scale_factor=(2.0, 2.0), method="nearest")
0 commit comments