RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix the bug in temporal_compression data loader (#754)

This commit is contained in:
Rodrigo Reis 2026-03-25 11:16:44 -03:00 committed by GitHub
parent 7f3309b291
commit 54f4732c9b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats 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