You can deploy this model and start earning money today!

Xibanya/
sunset_city

Name: sunset_city
Availability: InStock

Views

No views yet

Sunset Cities

This is the Malevich ruDALL-E model finetuned on anime screenshots of big cities at sunset.

installation

pip install rudalle

How to use

Basic implementation to get a list of image data objects.

python

from translate import Translator
from rudalle import get_rudalle_model, get_tokenizer, get_vae
from rudalle.pipelines import generate_images

model = get_rudalle_model('Malevich', pretrained=True, fp16=True, device='cuda')
model.load_state_dict(torch.load(CHECKPOINT_PATH))
vae = get_vae().to('cuda')
tokenizer = get_tokenizer()
input_text = Translator(to_lang='ru').translate('city at sunset')
images, _ = generate_images(
        text=input_text,
        tokenizer=tokenizer, dalle=model, vae=vae,
        images_num=1,
        top_k=2048,
        top_p=0.95,
        temperature=1.0
    )

the Malevich model only recognizes input in Russian. If you're going to paste Cyrillic directly into the code rather than filter an English prompt through the translate API, you will need to put this at the top of the file:

python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

Xibanya/sunset_city

Sunset Cities

installation

How to use

Xibanya/
sunset_city