Work on additional image embedding methods. Finalized zipper resampler. It works amazing

2024-02-10 09:00:05 -07:00
parent a8481c1670
commit e074058faa
7 changed files with 261 additions and 47 deletions
--- a/extensions_built_in/sd_trainer/SDTrainer.py
+++ b/extensions_built_in/sd_trainer/SDTrainer.py
@@ -998,7 +998,7 @@ class SDTrainer(BaseSDTrainProcess):
                        tensors_0_1=clip_images,
                        is_training=True,
                        has_been_preprocessed=True,
-                        quad_count=quad_count
+                        quad_count=quad_count,
                    )

                with self.timer('encode_prompt'):