put no_grad decorator on make_image closures (#375)

invoke-ai · Sep 5, 2022 · a20113d · a20113d
1 parent 0f93dad
commit a20113d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py
@@ -474,6 +474,7 @@ def _txt2img(
 
         sampler = self.sampler
 
+        @torch.no_grad()
         def make_image(x_T):
             uc, c = self._get_uc_and_c(prompt, skip_normalize)
             shape = [
@@ -528,6 +529,7 @@ def _img2img(
 
         t_enc = int(strength * steps)
 
+        @torch.no_grad()
         def make_image(x_T):
             uc, c = self._get_uc_and_c(prompt, skip_normalize)