New image generation img2img. various tweaks and fixes

2024-07-24 04:13:41 -06:00
parent 8d799031cf
commit 80aa2dbb80
9 changed files with 285 additions and 47 deletions
--- a/extensions_built_in/sd_trainer/SDTrainer.py
+++ b/extensions_built_in/sd_trainer/SDTrainer.py
@@ -483,6 +483,7 @@ class SDTrainer(BaseSDTrainProcess):
            noise=noise,
            sd=self.sd,
            unconditional_embeds=unconditional_embeds,
+            scaler=self.scaler,
            **kwargs
        )