Skip to content

ai-forever/KandiSuperRes

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

40 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

KandiSuperRes - diffusion model for super resolution

KandiSuperRes Flash Post | KandiSuperRes Post | Hugging Face Spaces | Telegram-bot | Our text-to-image model

KandiSuperRes Flash

Description

KandiSuperRes Flash is a new version of the diffusion model for super resolution. This model includes a distilled version of the KandiSuperRes model and a distilled model Kandinsky 3.0 Flash. KandiSuperRes Flash not only improves image clarity, but also corrects artifacts, draws details, improves image aesthetics. And one of the most important advantages is the ability to use the model in the "infinite super resolution" mode. 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.12;
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, flash=True, scale=2)

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

Infinite super resolution

With KandiSuperRes Flash you can infinitely enlarge images to x16 and more.

KandiSuperRes

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.

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, flash=False, scale=4)

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

Authors