Bug Fixes

2024-05-17 08:41:20 -06:00
parent 5a45c709cd
commit 68b7e159bc
4 changed files with 12 additions and 4 deletions
--- a/extensions_built_in/sd_trainer/SDTrainer.py
+++ b/extensions_built_in/sd_trainer/SDTrainer.py
@@ -1086,6 +1086,7 @@ class SDTrainer(BaseSDTrainProcess):
                        is_training=True,
                        has_been_preprocessed=True,
                        quad_count=quad_count,
+                        batch_size=noisy_latents.shape[0]
                    )

                with self.timer('encode_prompt'):