"PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16)
>>> # Enable memory optimizations.
>>> pipe.enable_model_cpu_offload()

>>> prompt = "A small cactus with a happy face in the Sahara desert."
>>> image = pipe(prompt).images[0]  classify_height_width_bin  < source > ( height: int width: int ratios: dict  )    Returns binned height and width.  encode_prompt  < source > ( prompt: Union do_classifier_free_guidance: bool = True negative