Created
November 25, 2022 22:33
-
-
Save chavinlo/e58b1314791783a748752142e7a3c5eb to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/bin/bash | |
#Install deps | |
apt-get update -y | |
apt-get install htop screen psmisc python3-pip unzip wget gcc g++ nano -y | |
#Install Python deps | |
wget https://gist.githubusercontent.com/chavinlo/fe8afc02e03d9cc4eb545c4c306c8a73/raw/d9a5ad446fe662dc3e6597163a1f8d5546a8a795/requirements.txt | |
pip install -r requirements.txt OmegaConf | |
pip install triton==2.0.0.dev20221120 | |
conda install xformers -c xformers/label/dev | |
#Download scripts | |
#trainer script | |
wget https://raw.githubusercontent.com/harubaru/waifu-diffusion/main/trainer/diffusers_trainer.py -O train.py | |
echo 'torchrun --nproc_per_node=1 train.py --model runwayml/stable-diffusion-v1-5 --run_name testrun1 --dataset datatrain --batch_size 3 --gradient_checkpointing="true" --use_8bit_adam="false" --fp16="true" --resize="true" --wandb="true" --project_id="icons" --use_xformers="true" --no_migration="true" --image_log_steps 300 --save_steps 1000' > run.sh | |
touch worked |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment