es_img).images[0]
>>> upscaled_image.save("upsampled_cat.png")

enable_attention_slicing

<
source
>
(
slice_size: typing.Union[str, int, NoneType] = 'auto'

)


Parameters 

slice_size (str or int, optional, defaults to "auto") —
When "auto", halves the input to the attention heads, so attention will be computed in two steps. If
"max", maxium amount of memory will be saved by running only one slice at a time. If a number is
provided, uses as m