Spaces:

baulab
/

Erasing-Concepts-In-Diffusion

Runtime error

RohitGandikota commited on Mar 20, 2023

Commit

6a74362

•

1 Parent(s): 8115240

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -189,11 +189,11 @@ class Demo:
                 iteration = int(iteration / self.nsteps * 1000)
-                positive_latents = self.diffuser.predict_noise(iteration, latents_steps[0], positive_text_embeddings, guidance_scale=3)
-                neutral_latents = self.diffuser.predict_noise(iteration, latents_steps[0], neutral_text_embeddings, guidance_scale=3)
             with finetuner:
-                negative_latents = self.diffuser.predict_noise(iteration, latents_steps[0], positive_text_embeddings, guidance_scale=3)
             positive_latents.requires_grad = False
             neutral_latents.requires_grad = False

                 iteration = int(iteration / self.nsteps * 1000)
+                positive_latents = self.diffuser.predict_noise(iteration, latents_steps[0], positive_text_embeddings, guidance_scale=1)
+                neutral_latents = self.diffuser.predict_noise(iteration, latents_steps[0], neutral_text_embeddings, guidance_scale=1)
             with finetuner:
+                negative_latents = self.diffuser.predict_noise(iteration, latents_steps[0], positive_text_embeddings, guidance_scale=1)
             positive_latents.requires_grad = False
             neutral_latents.requires_grad = False