is placed on the GPU while the others wait on the CPU. Components like the UNet that run for multiple iterations stay on the GPU until they’re no longer needed. Enable model offloading by calling enable_model_cpu_offload() on the pipeline:   Copied import torch