Karlo V1 Alpha Image Variations

🚀 Karlo v1 alpha

Karlo is a text-conditional image generation model. It's based on OpenAI's unCLIP architecture and improves the standard super-resolution model from 64px to 256px. It can recover high-frequency details in just a few denoising steps.

• Original codebase

🚀 Quick Start

Karlo is available in diffusers!

pip install diffusers transformers accelerate safetensors

💻 Usage Examples

Basic Usage

Text to image

from diffusers import UnCLIPPipeline
import torch

pipe = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha", 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("./frog.png")

Image variation

from diffusers import UnCLIPImageVariationPipeline
import torch
from PIL import Image

pipe = UnCLIPImageVariationPipeline.from_pretrained("kakaobrain/karlo-v1-alpha-image-variations", torch_dtype=torch.float16)
pipe = pipe.to('cuda')

image = Image.open("./frog.png")

image = pipe(image).images[0]

image.save("./frog-variation.png")

📚 Documentation

Model Architecture

Overview

Karlo is a text-conditional diffusion model based on unCLIP, composed of prior, decoder, and super-resolution modules. In this repository, we include the improved version of the standard super-resolution module for upscaling 64px to 256px only in 7 reverse steps, as illustrated in the figure below:

Specifically, the standard SR module trained by the DDPM objective upscales 64px to 256px in the first 6 denoising steps based on the respacing technique. Then, the additional fine-tuned SR module trained by VQ - GAN-style loss performs the final reverse step to recover high-frequency details. We find that this approach is very effective for upscaling low-resolution images in a small number of reverse steps.

Details

We train all components from scratch on 115M image-text pairs, including COYO - 100M, CC3M, and CC12M. For the Prior and Decoder, we use ViT - L/14 provided by OpenAI’s CLIP repository. Unlike the original implementation of unCLIP, we replace the trainable transformer in the decoder with the text encoder in ViT - L/14 for efficiency. For the SR module, we first train the model using the DDPM objective in 1M steps, followed by an additional 234K steps to fine - tune the additional component. The table below summarizes the important statistics of our components:

In the checkpoint links, ViT - L/14 is equivalent to the original version, but we include it for convenience. We also note that ViT - L/14 - stats is required to normalize the outputs of the prior module.

Evaluation

We quantitatively measure the performance of Karlo - v1.0.alpha in the validation split of CC3M and MS - COCO. The table below presents the CLIP - score and FID. To measure FID, we resize the shorter side of the image to 256px and then crop it at the center. We set the classifier - free guidance scales for the prior and decoder to 4 and 8 in all cases. We find that our model achieves reasonable performance even with 25 sampling steps of the decoder.

CC3M

MS - COCO

For more information, please refer to the upcoming technical report.

Training Details

This alpha version of Karlo is trained on 115M image - text pairs, including the [COYO](https://github.com/kakaobrain/coyo - dataset) - 100M high - quality subset, CC3M, and CC12M. For those interested in a better version of Karlo trained on more large - scale high - quality datasets, please visit the landing page of our application B^DISCOVER.

📄 License

The license for this project is creativeml - openrail - m.

🔗 BibTex

If you find this repository useful in your research, please cite:

@misc{kakaobrain2022karlo-v1-alpha,
  title         = {Karlo-v1.0.alpha on COYO-100M and CC15M},
  author        = {Donghoon Lee, Jiseob Kim, Jisu Choi, Jongmin Kim, Minwoo Byeon, Woonhyuk Baek and Saehoon Kim},
  year          = {2022},
  howpublished  = {\url{https://github.com/kakaobrain/karlo}},
}