lama-cleaner обработка фото, нейронка. Py + Docker + GPU + CPU

Lama Cleaner

Свободный инструмент для рисования с открытым исходным кодом, основанный на модели SOTA AI.

Позволяет:
Удалить все лишнее с изображения
Исправить старое фото
Заменить что-то на изображении

перерисовка действует в режиме:

https://github.com/Sanster/lama-cleaner#inpainting-strategy

sudo docker run -p 8081:8081 -v /path/to/torch_cache:/root/.cache/torch -v /path/to/huggingface_cache:/root/.cache/huggingface --rm cwq1913/lama-cleaner:cpu-0.24.4 lama-cleaner --device=cpu --port=8081 --host=0.0.0.0 --hf_access_token=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

sd1.5 модель требует токен, для этого нужно запустить контейнер с токеном. Обработка на CPU ведется около 20 минут, вырезание дефолтными моделями 3 сек.