diff --git a/studio/backend/core/inference/diffusion_vae_quant.py b/studio/backend/core/inference/diffusion_vae_quant.py index 0bac932f4e..b9acefd863 100644 --- a/studio/backend/core/inference/diffusion_vae_quant.py +++ b/studio/backend/core/inference/diffusion_vae_quant.py @@ -198,7 +198,6 @@ def _vae_conv_ndims(vae: Any) -> tuple[int, ...]: ndims: set[int] = set() try: from torch import nn - for module in vae.modules(): if isinstance(module, nn.Conv3d): ndims.add(3)