layer zeroing

2024-05-13 12:01:44 +03:00
parent 0df506ee4a
commit a56b5f0cd6
1 changed files with 1 additions and 0 deletions
@@ -1018,6 +1018,7 @@ def add_brushnet_patch(model, brushnet, torch_dtype, conditioning_latents,
                h += self.brushnet_sample.to(h.dtype).to(h.device)
            else:
                h += self.brushnet_sample
+            self.brushnet_sample = 0
        return h

    for i, block in enumerate(model.model.diffusion_model.input_blocks):