enable_attention_slicing()
>>> image = pipe(prompt).images[0]  enable_model_cpu_offload  < source > ( gpu_id: typing.Optional[int] = None device: typing.Union[torch.device, str] = '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 accelerator that shall be