sdxs3d / README.md
recoilme's picture
2409
9cc90fc
metadata
license: apache-2.0

train

update:

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

pip + torch:

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

repo:

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

dataset:

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

auth:

huggingface-cli login
wandb login

run it!

nohup accelerate launch train.py &