fix xformers

app.py

@@ -11,7 +11,7 @@ from transformers import CLIPModel, CLIPProcessor
 
 from tools import synth
 
-device = "cuda"  # if torch.cuda.is_available() else "cpu"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 model_path = "runwayml/stable-diffusion-v1-5"
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14").to(device)
 clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")

tools/synth.py

@@ -169,6 +169,7 @@ def pipe_img(
         )  # lower is faster but lower quality
         helper.enable()
         if torch.cuda.is_available():
+            pipe.to("cuda")
             pipe.enable_xformers_memory_efficient_attention()
         if use_torchcompile:
             pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)