sdxs / TRAIN.md
recoilme's picture
Fresh start
94a2309
---
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 &
```