Let's make sure it runs on GPU :-) (#3)

- Let's make sure it runs on GPU :-) (3ad69a045e743950cb8347367af8309db25ecb4f)


Co-authored-by: Patrick von Platen <[email protected]>

app.py

@@ -2,6 +2,7 @@ from diffusers import StableDiffusionPipeline
 import gradio as gr
 
 pipe = StableDiffusionPipeline.from_pretrained("nitrosocke/Arcane-Diffusion")
+pipe = pipe.to("cuda")
 
 def inference(prompt, guidance, steps):
     all_images = []