xformers

app.py

@@ -95,11 +95,12 @@ def load_pipeline():
         IMAGE_MODEL_REPO_ID,
         vae=vae,
         torch_dtype=torch.float16,
+        use_safetensors=True,
         add_watermarker=False,
         custom_pipeline="lpw_stable_diffusion_xl",
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    pipe.unet.set_attn_processor(AttnProcessor2_0())
+    pipe.enable_xformers_memory_efficient_attention()
     if CPU_OFFLOAD:  # local
         pipe.enable_sequential_cpu_offload(gpu_id=0, device=device)
     else:

requirements.txt

@@ -1,7 +1,7 @@
-accelerate
+torch==2.4.0
 
-diffusers
-torch
-transformers
-xformers
-spaces
+accelerate==0.34.2
+diffusers==0.30.1
+transformers==4.45.1
+spaces==0.30.2
+xformers