Initial Commit

app.py

@@ -6,8 +6,7 @@ model_name = "runwayml/stable-diffusion-v1-5"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 if device == "cuda":
-    pipe = StableDiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float16)
-    pipe = pipe.to(device)
+    pipe = StableDiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float16).to(device)
 else:
     pipe = StableDiffusionPipeline.from_pretrained(model_name).to(device)