nvidia version created

apple_silicon.py

@@ -0,0 +1,12 @@
+from diffusers import StableDiffusionPipeline
+
+DEVICE = 'mps'
+
+pipe = StableDiffusionPipeline.from_pretrained(
+    "CompVis/stable-diffusion-v1-4",
+    use_auth_token=True
+).to(DEVICE)
+prompt = "Synthwave moonlight"
+image = pipe(prompt).images[0]
+image.save("pictures/Synthwave.png")
+

test.py → nvidia.py

@@ -7,9 +7,9 @@ pipe = StableDiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",
     use_auth_token=True
 ).to(DEVICE)
-prompt = "Ezreal on a moon background"
+prompt = "Barcelona moonlight"
 pipe.enable_attention_slicing()
 # _ = pipe(prompt, num_inference_steps=1)
 image = pipe(prompt).images[0]
-image.save("Ezreal1.png")
+image.save("pictures/Barcelona1.png")