KandiSuperRes - diffusion model for 4K super resolution
Habr Post | Github | 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
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
- Downloads last month
- 0
Unable to determine this model's library. Check the
docs
.