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

Fix the bug in temporal_compression data loader (#754)

Browse Source
This commit is contained in:
Rodrigo Reis
2026-03-25 11:16:44 -03:00
committed by GitHub
parent 7f3309b291
commit 54f4732c9b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/data_loader.py
View File

@@ -508,9 +508,9 @@ class AiToolkitDataset(LatentCachingMixin, ControlCachingMixin, CLIPCachingMixin
temporal_compression = 8
if self.sd is not None:
if hasattr(self.sd.vae.config, 'scale_factor_temporal'):
if hasattr(self.sd.vae, 'config') and hasattr(self.sd.vae.config, 'scale_factor_temporal'):
temporal_compression = self.sd.vae.config.scale_factor_temporal
if hasattr(self.sd.unet.config, 'temporal_compression_ratio'):
if hasattr(self.sd.unet, 'config') and hasattr(self.sd.unet.config, 'temporal_compression_ratio'):
temporal_compression = self.sd.unet.config.temporal_compression_ratio
bad_count = 0