diff --git a/src/diffusers/models/autoencoders/autoencoder_kl_qwenimage21.py b/src/diffusers/models/autoencoders/autoencoder_kl_qwenimage21.py index ac6af47ab493..665f9377648d 100644 --- a/src/diffusers/models/autoencoders/autoencoder_kl_qwenimage21.py +++ b/src/diffusers/models/autoencoders/autoencoder_kl_qwenimage21.py @@ -515,7 +515,7 @@ def __init__(self, in_dim, out_dim, dropout, num_res_blocks, temperal_downsample def forward(self, x, feat_cache=None, feat_idx=None) -> torch.Tensor: if feat_idx is None: feat_idx = [0] - x_copy = x.clone() + x_copy = x for resnet in self.resnets: x = resnet(x, feat_cache=feat_cache, feat_idx=feat_idx) if self.downsampler is not None: @@ -714,7 +714,7 @@ def forward(self, x, feat_cache=None, feat_idx=None, first_chunk=False) -> torch Returns: torch.Tensor: Output tensor """ - x_copy = x.clone() + x_copy = x for resnet in self.resnets: if feat_cache is not None: diff --git a/src/diffusers/models/autoencoders/autoencoder_kl_wan.py b/src/diffusers/models/autoencoders/autoencoder_kl_wan.py index dcb280ff21a2..e05c0a1c18cd 100644 --- a/src/diffusers/models/autoencoders/autoencoder_kl_wan.py +++ b/src/diffusers/models/autoencoders/autoencoder_kl_wan.py @@ -501,7 +501,7 @@ def __init__(self, in_dim, out_dim, dropout, num_res_blocks, temperal_downsample self.downsampler = None def forward(self, x, feat_cache=None, feat_idx=[0]) -> torch.Tensor: - x_copy = x.clone() + x_copy = x for resnet in self.resnets: x = resnet(x, feat_cache=feat_cache, feat_idx=feat_idx) if self.downsampler is not None: @@ -700,7 +700,7 @@ def forward(self, x, feat_cache=None, feat_idx=[0], first_chunk=False) -> torch. Returns: torch.Tensor: Output tensor """ - x_copy = x.clone() + x_copy = x for resnet in self.resnets: if feat_cache is not None: