fix to work caching latents #1696

kohya-ss · Oct 18, 2024 · d8d7142 · d8d7142
1 parent 2500f5a
commit d8d7142
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdxl_train_control_net_lllite.py b/sdxl_train_control_net_lllite.py
@@ -181,7 +181,7 @@ def train(args):
         vae.requires_grad_(False)
         vae.eval()
 
-        train_dataset_group.new_cache_latents(vae, accelerator.is_main_process)
+        train_dataset_group.new_cache_latents(vae, accelerator)
 
         vae.to("cpu")
         clean_memory_on_device(accelerator.device)