Create midjourney.py

midjourney.py

@@ -0,0 +1,8 @@
+from diffusers import StableDiffusionPipeline
+import torch
+model_id = "prompthero/openjourney"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = pipe.to("cuda")
+prompt = "retro serie of different cars with different colors and shapes, mdjrny-v4 style"
+image = pipe(prompt).images[0]
+image.save("./retro_cars.png")