- Приложение Gradio для InfiniteYou (
infiniteyou_app.py) - Приложение Gradio для KV-Edit (
kvedit_app.py)
- GPU: Рекомендуется NVIDIA A40 45GB (InfiniteYou требует мин. 43GB).
- Хранилище: ~120GB для чекпоинтов.
- Окружение Python (см.
requirements.txt).
- Клонировать:
git clone --recurse-submodules https://github.com/Kiberchaika/HeadSwapDiT && cd HeadSwapDiT # Если клонировали без --recurse-submodules, выполните: git submodule update --init --recursive
- (Только Linux) Системные зависимости: (например, для OpenCV)
sudo apt-get update && sudo apt-get install -y libgl1-mesa-glx - Окружение Python:
python -m venv venv source venv/bin/activate # Windows: venv\Scripts\activate pip install -r requirements.txt
- Настройка модели (InfiniteYou - FLUX.1-dev):
- Доступ: Запросите доступ на huggingface.co/black-forest-labs/FLUX.1-dev.
- Аутентификация: Войдите через CLI с помощью
huggingface-cli login, используя токен чтения (сгенерировать здесь). - Альтернатива: Скачайте файлы модели вручную после получения доступа и поместите их в ожидаемый каталог (например,
./models/FLUX.1-dev).
- Активировать окружение:
source venv/bin/activate - Запустить приложения Gradio:
- InfiniteYou:
python infiniteyou_app.py - KV-Edit:
python kvedit_app.py
- InfiniteYou:
- Vast.ai Termux: Отключите автозапуск с помощью
touch ~/.no_auto_tmuxв домашнем каталоге и переподключитесь.