Created
November 24, 2022 18:06
-
-
Save chavinlo/432a1f0101e0ed4e4eb123aff42ff927 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/41e062890b91cde16ac146719a669308/raw/9ebca64f70379a6012c804e4a39ede1db4a6b665/requirements.txtpip install -r requirements.txt OmegaConf | |
pip install triton==2.0.0.dev20221120 | |
conda install xformers -c xformers/label/dev | |
#Download weights | |
wget https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-full-opt.ckpt | |
#Download scripts | |
#trainer script | |
wget https://gist.githubusercontent.com/chavinlo/3fd36daeb05d9a578e5150aa812caa8d/raw/18d0d9a13a2bdd9455f8dff39bb5d92991484786/diffusers_trainer_nodatasync.py -O train.py | |
#convert script | |
wget https://raw.githubusercontent.com/huggingface/diffusers/d9cfe325a53502641f16ce4f839391c5b0d0a684/scripts/convert_original_stable_diffusion_to_diffusers.py | |
#convert | |
python3 convert_original_stable_diffusion_to_diffusers.py --checkpoint_path wd-v1-3-full-opt.ckpt --dump_path wdiffuser | |
echo 'torchrun --nproc_per_node=1 train.py --model wdiffuser --run_name testrun1 --dataset datatrain --batch_size 3 --gradient_checkpointing="true" --use_8bit_adam="false" --fp16="true" --resize="true" --wandb="true" --project_id="hivemind_experiments" --use_xformers="true" --no_migration="true" --image_log_steps 500 --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