setup.sh

#!/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://gist.githubusercontent.com/chavinlo/1d98223425aea0309ae7973e56c3d213/raw/0caabb3ea258f09e953f16040d2cd18a1c60dbb7/diffusers_hivemind_final.py -O train.py

echo 'torchrun --nproc_per_node=1 train.py --workingdirectory hivemindtemp --wantedimages 500 --datasetserver SERVERGOESHERE --node="false" --model runwayml/stable-diffusion-v1-5 --run_name testrun1 --batch_size 2 --gradient_checkpointing="true" --use_8bit_adam="false" --fp16="true" --resize="true" --wandb="false" --use_xformers="true" --no_migration="true" --image_log_steps 400 --save_steps 1000' > run.sh

touch worked