Added additional config options for custom plugins I needed

2024-01-15 08:31:09 -07:00
parent e190fbaeb8
commit 5276975fb0
7 changed files with 37 additions and 31 deletions
--- a/toolkit/guidance.py
+++ b/toolkit/guidance.py
@@ -413,6 +413,7 @@ def get_guided_loss_polarity(
    device = sd.device_torch
    with torch.no_grad():
        dtype = get_torch_dtype(dtype)
+        noise = noise.to(device, dtype=dtype).detach()

        conditional_latents = batch.latents.to(device, dtype=dtype).detach()
        unconditional_latents = batch.unconditional_latents.to(device, dtype=dtype).detach()