make it even clearer

ZeroCool940711 · May 13, 2023 · 3a987f4 · 3a987f4
1 parent 02afc54
commit 3a987f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/muse_maskgit_pytorch/trainers/maskgit_trainer.py b/muse_maskgit_pytorch/trainers/maskgit_trainer.py
@@ -109,7 +109,7 @@ def save_validation_images(
 
         if self.accelerator.is_main_process:
             save_image(images, save_file, "png")
-            self.log_validation_images([Image.open(save_file)], step, ["\n\n".join(validation_prompts)])
+            self.log_validation_images([Image.open(save_file)], step, ["\n---\n".join(validation_prompts)])
         return save_file
 
     def train(self):