test

app.py

@@ -39,7 +39,7 @@ def ui():
         device=device,
         hf_cache_dir=CACHE_DIR,
         use_safetensors=True,
-        enable_xformers_memory_efficient_attention=True,
+        enable_xformers_memory_efficient_attention=torch.cuda.is_available(),
     )
 
     preprocessors = ['canny']