ai-forever / KandiSuperRes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

KandiSuperRes - diffusion model for 4K super resolution

Habr Post | Hugging Face Spaces | Telegram-bot | Technical Report| Our text-to-image model

Description

KandiSuperRes is an open-source diffusion model for x4 super resolution. This model is based on the Kandinsky 3.0 architecture with some modifications. For generation in 4K, the MultiDiffusion algorithm was used, which allows to generate panoramic images. For more information: details of architecture and training, example of generations check out our Habr post.

Installing

To install repo first one need to create conda environment:

conda create -n kandisuperres -y python=3.8;
source activate kandisuperres;
pip install -r requirements.txt;

How to use

Check our jupyter notebook KandiSuperRes.ipynb with example.

from KandiSuperRes import get_SR_pipeline
from PIL import Image

sr_pipe = get_SR_pipeline(device='cuda', fp16=True)

lr_image = Image.open('')
sr_image = sr_pipe(lr_image)

Authors

About

License:Apache License 2.0


Languages

Language:Jupyter Notebook 99.9%Language:Python 0.1%