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

DEF for fake vae and adjust scaling

Browse Source
This commit is contained in:
Jaret Burkett committed 2025-09-12 18:09:08 -06:00
1 parent b95c17dc17
commit 3666b112a8
2 files changed
+5 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

toolkit/models/diffusion_feature_extraction.py
+3 -3
View File
@@ -531,9 +531,9 @@ class DiffusionFeatureExtractor4(nn.Module):
stepped_latents = torch.cat(stepped_chunks, dim=0)
latents = stepped_latents.to(self.vae.device, dtype=self.vae.dtype)
scaling_factor = self.vae.config['scaling_factor'] if 'scaling_factor' in self.vae.config else 1.0
shift_factor = self.vae.config['shift_factor'] if 'shift_factor' in self.vae.config else 0.0
scaling_factor = self.vae.config.scaling_factor if hasattr(self.vae.config, 'scaling_factor') else 1.0
shift_factor = self.vae.config.shift_factor if hasattr(self.vae.config, 'shift_factor') else 0.0
latents = (latents / scaling_factor) + shift_factor
if is_video:
# if video, we need to unsqueeze the latents to match the vae input shape