Merge branch 'sd3' into diff_output_prsv

kohya-ss · Oct 19, 2024 · 09b4d1e · 09b4d1e
2 parents 2c45d97 + d8d7142
commit 09b4d1e
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)