  guidance_scale=0,
...     generator=generator,
... ).images[0]

>>> upscaled_image.save("../images/a2.png")  enable_sequential_cpu_offload  < source > ( gpu_id: Optional = None device: Union = 'cuda'  )   Parameters   gpu_id (int, optional) —
The ID of the accelerator that shall be used in inference. If not specified, it will default to 0.   device (torch.Device or str, optional, defaults to “cuda”) —
The PyTorch device type of the accelerato