update

2026-10-23 08:15:18 +00:00 · 2024-12-21 15:18:46 +01:00
parent 2da8cb5798
commit 0db7bf28aa
3 changed files with 159 additions and 30 deletions
@@ -319,7 +319,7 @@ class Trainer:
                generator=self.state.generator,
                precompute=True,
            )
-            filename = latents_dir / f"latents-{i}-{index}.pt"
+            filename = latents_dir / f"latents-{self.state.accelerator.process_index}-{index}.pt"
            torch.save(latent_conditions, filename.as_posix())
            index += 1
            progress_bar.update(1)