from diffusers import UnCLIPPipeline
import torch
import random
import numpy as np
pipe = UnCLIPPipeline.from_pretrained("fusing/karlo_unclip", torch_dtype=torch.float16)
pipe = pipe.to('cuda')
prompt = "a high-resolution photograph of a big red frog on a green leaf."
image = pipe([prompt]).images[0]
image.save("/home/patrick_huggingface_co/images/frog.png")
