Nikita Balabin

Results 1 comments of Nikita Balabin

``` from diffusers import PixArtAlphaPipeline, Transformer2DModel import torch pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16).to('cuda') prompts = ["a man", "cat riding a bike in the forest, watercolor style"] images = pipe(prompt=prompts, num_images_per_prompt=2).images ```...