From 7861f92b13787f5460978a463c8737ce235a1455 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Fri, 23 Oct 2026 04:40:42 -0600 Subject: [PATCH] Remove the forced no grad on vae decode for moss so it can be used with DFEs --- extensions_built_in/diffusion_models/moss_image/moss_image.py | 1 - 1 file changed, 1 deletion(-) diff --git a/extensions_built_in/diffusion_models/moss_image/moss_image.py b/extensions_built_in/diffusion_models/moss_image/moss_image.py index c5b67d7..c01d870 100644 --- a/extensions_built_in/diffusion_models/moss_image/moss_image.py +++ b/extensions_built_in/diffusion_models/moss_image/moss_image.py @@ -629,7 +629,6 @@ class MossImageModel(BaseModel): batch = torch.stack([img.to(device, dtype=self.vae.dtype) for img in image_list]) return self.vae.encode(batch).to(device, dtype=dtype) - @torch.no_grad() def decode_latents(self, latents: torch.Tensor, device=None, dtype=None): if device is None: device = self.vae_device_torch