 --enable_xformers_memory_efficient_attention 

Inference
	
Once you have trained a model using the above command, you can run inference using the below command. Make sure to include the modifier token (e.g. \<new1> in above example) in your prompt.


	Copied
import torch
from diffusers import DiffusionPipeline

pipe = DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16).to("cuda")
pipe.unet.load_attn_pr