ru dalle

Rudall-e

텍스트에서 이미지를 생성합니다

 pip install rudalle==1.1.3

? HF 모델 :

Rudall-e Malevich (XL)
Rudall-e Emojich (xl) (여기 읽기)
Rudall-e 초현실 주의자 (XL)
Rudall-e Kandinsky (XXL) (곧)

최소 예 :

예제 사용법 3.5GB VRAM을 가진 Rudall-e Malevich (XL)!

미세 조정 예

Rudalle의 세대 :

 import ruclip
from rudalle . pipelines import generate_images , show , super_resolution , cherry_pick_by_ruclip
from rudalle import get_rudalle_model , get_tokenizer , get_vae , get_realesrgan
from rudalle . utils import seed_everything

# prepare models:
device = 'cuda'
dalle = get_rudalle_model ( 'Malevich' , pretrained = True , fp16 = True , device = device )
tokenizer = get_tokenizer ()
vae = get_vae ( dwt = True ). to ( device )

# pipeline utils:
realesrgan = get_realesrgan ( 'x2' , device = device )
clip , processor = ruclip . load ( 'ruclip-vit-base-patch32-384' , device = device )
clip_predictor = ruclip . Predictor ( clip , processor , device , bs = 8 )
text = 'радуга на фоне ночного города'

seed_everything ( 42 )
pil_images = []
scores = []
for top_k , top_p , images_num in [
    ( 2048 , 0.995 , 24 ),
]:
    _pil_images , _scores = generate_images ( text , tokenizer , dalle , vae , top_k = top_k , images_num = images_num , bs = 8 , top_p = top_p )
    pil_images += _pil_images
    scores += _scores

show ( pil_images , 6 )