Spaces:

Pusheen
/

LoCo

Sleeping

Pusheen commited on Feb 26

Commit

a89da3b

•

1 Parent(s): bfc1c93

Update gligen/ldm/models/diffusion/plms.py

Files changed (1) hide show

gligen/ldm/models/diffusion/plms.py CHANGED Viewed

@@ -209,6 +209,7 @@ class PLMSSampler(object):
             x = x - grad_cond
             x = x.detach()
             iteration += 1
         return x

             x = x - grad_cond
             x = x.detach()
             iteration += 1
+            torch.cuda.empty_cache()
         return x