flush() Notice that while initializing pipe, you’re setting text_encoder to None so that it’s not loaded. Once the latents are computed, pass it off to the VAE to decode into a real image:   Copied with torch.no_grad():