Create run.py

run.py

@@ -0,0 +1,21 @@
+import torch
+from diffusers import StableDiffusionPipeline
+
+# Load the Stable Diffusion pipeline
+pipeline = StableDiffusionPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5", 
+    torch_dtype=torch.float16
+)
+pipeline.to("cuda")  # Use the GPU
+
+# Load your safetensors weights
+pipeline.unet.load_attn_procs("v1-5-pruned-emaonly.safetensors")
+
+# Generate an image
+prompt = "A serene landscape with mountains and a lake at sunset"
+result = pipeline(prompt, guidance_scale=7.5)  # Adjust guidance scale if needed
+image = result.images[0]
+
+# Save and display the image
+image.save("generated_image.png")
+image.show()