diff --git a/comfy/sd.py b/comfy/sd.py index ca268cbbe..12f1373c5 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -820,6 +820,7 @@ class VAE: do_tile = True if do_tile: + torch.cuda.empty_cache() dims = samples_in.ndim - 2 if dims == 1 or self.extra_1d_channel is not None: pixel_samples = self.decode_tiled_1d(samples_in) @@ -895,6 +896,7 @@ class VAE: do_tile = True if do_tile: + torch.cuda.empty_cache() if self.latent_dim == 3: tile = 256 overlap = tile // 4