various bug fixes. Created an contextual alpha mask module to calculate alpha mask

2024-01-18 16:34:27 -07:00
parent 86c70a2a1f
commit f17ad8d794
7 changed files with 93 additions and 28 deletions
--- a/extensions_built_in/sd_trainer/SDTrainer.py
+++ b/extensions_built_in/sd_trainer/SDTrainer.py
@@ -1085,7 +1085,7 @@ class SDTrainer(BaseSDTrainProcess):
                            noise=noise,
                            batch=batch,
                            unconditional_embeds=unconditional_embeds
-                        )
+                        ).detach()

                # do the custom adapter after the prior prediction
                if self.adapter and isinstance(self.adapter, CustomAdapter) and has_clip_image: