Skip to content

Commit 211f58f

Browse files
committed
Fix error
1 parent 557ee7d commit 211f58f

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

src/maxdiffusion/models/wan/autoencoder_kl_wan.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,6 @@ class WanResample(nnx.Module):
177177
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):
178178
self.dim = dim
179179
self.mode = mode
180-
self.time_conv = None
181180

182181
if mode == "upsample2d":
183182
self.upsample = WanUpsample(scale_factor=(2.0, 2.0), method="nearest")

0 commit comments

Comments
 (0)