Prompt:
little cute gremlin wearing a jacket, cinematic, vivid colors, intricate masterpiece, golden ratio, highly detailed
Image:
(Image from https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png)
Prompt:
cat wizard, gandalf, lord of the rings, detailed, fantasy, cute, adorable, Pixar, Disney, 8k
This model requires additional module.
pip3 install transformers
Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.
For the sample image, (Text-to-image)
$ python3 sdxl-turbo.py
If you want to specify the input prompt, put the prompt after the --input
option.
You can use --savepath
option to change the name of the output file to save.
$ python3 sdxl-turbo.py --input PROMPT --savepath SAVE_WAV_PATH
For Image-to-image, specify the input image file after the --init_image
option.
$ python3 sdxl-turbo.py --input "cat wizard, gandalf, lord of the rings, detailed, fantasy, cute, adorable, Pixar, Disney, 8k" --init_image cat.png
Pytorch
ONNX opset=14
unet.onnx.prototxt
text_encoder.onnx.prototxt
text_encoder_2.onnx.prototxt
vae_encoder.onnx.prototxt
vae_decoder.onnx.prototxt