Install:
git clone https://huggingface.co/ontocord/minidalle/
pip install transformers sentencepiece
Load model:
from minidalle.modeling_minidalle import *
import torch
minidalle = DalleModel.from_pretrained("minidalle").eval().cpu()
Use:
text="""A woman riding a black horse next to a blue fence in central park"""
img = minidalle.generate(
text=text,
image_output=True,
token_output=False
)