Spaces:

wondervictor
/

ControlAR

Running on Zero

wondervictor commited on Dec 11, 2024

Commit

e57f3c4

verified ·

1 Parent(s): 340cc7b

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -279,7 +279,6 @@ class Model:
                 sample.permute(1, 2, 0).cpu().detach().numpy().clip(0, 255).astype(np.uint8))
             for sample in samples
         ]
-        del image_tensor
         del condition_img
         torch.cuda.empty_cache()
         return samples

                 sample.permute(1, 2, 0).cpu().detach().numpy().clip(0, 255).astype(np.uint8))
             for sample in samples
         ]
         del condition_img
         torch.cuda.empty_cache()
         return samples