Данный алгоритм основан на Stable Diffusion.
Для запуска алгоритма потребуется:
- Скачать репозиторий
- Скачать модели по ссылке: https://drive.google.com/drive/folders/1SQDBt4YekymAVUmfnKmyXYtAM9UD-B1N?usp=sharing. Папки unet, vae, image_encoder и text_encoder необходимо сохранить в папке pretrained_model (pretrained_model/unet, pretrained_model/vae, pretrained_model/image_encoder, pretrained_model/text_encoder).
- Установить зависимости: pip install -r requirements.txt
Инференс: python run.py /path/to/content_image /path/to/style_image /path/to/output/image.png(jpg)
Прогнанные изображения из тестового задания:
