sdxs / TRAIN.md
recoilme's picture
Fresh start
94a2309
metadata
license: apache-2.0

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

apt update
apt install git-lfs
git config --global credential.helper store

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

python -m pip install --upgrade pip
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 -U
pip install flash-attn --no-build-isolation # optional

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

git clone https://huggingface.co/AiArtLab/sdxs
cd sdxs/
pip install -r requirements.txt

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

mkdir datasets
cd datasets
huggingface-cli download AiArtLab/384 --local-dir 384 --repo-type dataset

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

huggingface-cli login
wandb login

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

nohup accelerate launch train.py &