evan/ai-toolkit
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove the forced no grad on vae decode for moss so it can be used with DFEs

Browse Source
This commit is contained in:
Jaret Burkett committed 2026-10-23 04:40:42 -06:00
1 parent 7cc9fef469
commit 7861f92b13
1 file changed
-1
Download Patch File Download Diff File Expand all files Collapse all files
extensions_built_in/diffusion_models/moss_image/moss_image.py
-1
View File
@@ -629,7 +629,6 @@ class MossImageModel(BaseModel):
batch = torch.stack([img.to(device, dtype=self.vae.dtype) for img in image_list]) batch = torch.stack([img.to(device, dtype=self.vae.dtype) for img in image_list])
return self.vae.encode(batch).to(device, dtype=dtype) return self.vae.encode(batch).to(device, dtype=dtype)
@torch.no_grad()
def decode_latents(self, latents: torch.Tensor, device=None, dtype=None): def decode_latents(self, latents: torch.Tensor, device=None, dtype=None):
if device is None: if device is None:
device = self.vae_device_torch device = self.vae_device_torch