metadata
license: mit
library_name: transformers
pipeline_tag: text-to-image
tags:
- text-to-image
- transformer
- custom_code
- EZFGraphic
datasets:
- OpenRussianAI/EZF-Dataset-Image
EZFGraphic
An example model of EZFGraphic, Txt2Img generate
Usage
import torch
from PIL import Image
from transformers import AutoModel
from huggingface_hub import hf_hub_download
from tokenizers import Tokenizer
model = AutoModel.from_pretrained("OpenRussianAI/ezfgraphic", trust_remote_code=True).eval()
tok = Tokenizer.from_file(hf_hub_download("OpenRussianAI/ezfgraphic", "tokenizer.json"))
def generate(text, ctx=32):
ids = tok.encode(text).ids[:ctx]
ids = ids + [0] * (ctx - len(ids))
with torch.no_grad():
pixels = model(torch.tensor([ids]))
return (pixels[0] * 0.5 + 0.5).clamp(0, 1)
img = generate("нарисуй красный круг на тёмном фоне")
img_np = (img.permute(1, 2, 0).numpy() * 255).astype("uint8")
Image.fromarray(img_np).save("output.png")
MIT