AiArtLab/sdxs

Краткая инструкция по установке Обновите систему и установите git-lfs:

apt update
apt install git-lfs

Обновите pip и установите требуемые пакеты:

python -m pip install --upgrade pip
pip install -r requirements.txt

Настройте git-credentials:

git config --global credential.helper store

Клонируйте репозиторий:

git clone https://huggingface.co/AiArtLab/sdxs
cd sdxs/

Подготовьте датасет:

mkdir datasets
cd datasets
git clone https://huggingface.co/datasets/AiArtLab/imagenet-1kk

Выполните вход в сервисы:

huggingface-cli login
wandb login

Настройте accelerate для многокарточного обучения:

accelerate config

Запустите обучение!

nohup accelerate launch train.py &