setup.sh

#!/bin/bash

#Install deps
apt-get update -y
apt-get install htop screen psmisc python3-pip unzip wget gcc g++ -y

#Install Python deps
wget https://gist.githubusercontent.com/chavinlo/04330ffe95223f7a0a42de81526199b7/raw/b1c19df0e694936e1dadc7082d2326d6382b5d15/requirements.txt
pip install -r requirements.txt OmegaConf 

#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 --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 --batchsize 1 --gradient_checkpointing="true" --use_8bit_adam="false" --fp16="true" --resize="true" --wandb="true" --project_id="hivemind_experiments" --use_xformers="False" --no_migration="true" --image_log_steps 500 --save_steps 1000' > run.sh

touch worked