 Workaround for not accepting attention shape using VAE for Flash Attention
>>> pipe.vae.enable_xformers_memory_efficient_attention(attention_op=None)  from_pretrained  < source > ( pretrained_model_name_or_path: typing.Union[str, os.PathLike, NoneType] **kwargs  )   Parameters   pretrained_model_name_or_path (str or os.PathLike, optional) —
Can be either:

A string, the repo id (for example CompVis/ldm-text2im-large-256) of a pretrained pipeli