Aigencloud / README.md
dpredrag's picture
Update README.md
060d1ed
|
raw
history blame
811 Bytes
metadata
inference: true
language:
  - en
tags:
  - stable-diffusion
  - text-to-image
license: creativeml-openrail-m

AiGenerator cloud open source Stable Diffusion fine tuned model on Midjourney images


Use prompt: 'mdjrny-v4 style'

Gradio

You can also export the model to ONNX, MPS and/or FLAX/JAX.

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")