From 54f4732c9b75a0211bd3187d440a5b2e6e88849a Mon Sep 17 00:00:00 2001 From: Rodrigo Reis Date: Wed, 25 Mar 2026 11:16:44 -0300 Subject: [PATCH] Fix the bug in temporal_compression data loader (#754) --- toolkit/data_loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/toolkit/data_loader.py b/toolkit/data_loader.py index dbfde714..2391adcd 100644 --- a/toolkit/data_loader.py +++ b/toolkit/data_loader.py @@ -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