einar/scripts
1
0
Fork 0
Code Releases Activity

Create folders and clean up data

Browse source
This commit is contained in:
Luca Beltrame 2023-06-04 15:15:28 +02:00
parent c9d3caf251
commit a38fdb1457
Signed by: einar
GPG key ID: 4707F46E9EC72DEC
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
stable-diffusion/setup_runpod_lora_learn.sh
View file

@ -20,6 +20,9 @@ pip install --upgrade \
echo "Installing xformers" echo "Installing xformers"
pip install --no-deps xformers==0.0.17 pip install --no-deps xformers==0.0.17
echo "Installing TensorRT"
pip install tensorrt pip install tensorrt
echo "Cloning the project..." echo "Cloning the project..."
@ -33,6 +36,8 @@ TEMP_REQUIREMENTS_FILE="${DIR}/requirements_tmp_for_setup.txt"
awk -v dir="$DIR" '/#.*kohya_ss.*library/{print; getline; sub(/^\.$/, dir)}1' "$DIR/requirements.txt" >"$TEMP_REQUIREMENTS_FILE" awk -v dir="$DIR" '/#.*kohya_ss.*library/{print; getline; sub(/^\.$/, dir)}1' "$DIR/requirements.txt" >"$TEMP_REQUIREMENTS_FILE"
python -m pip install --use-pep517 --upgrade -r "$TEMP_REQUIREMENTS_FILE" python -m pip install --use-pep517 --upgrade -r "$TEMP_REQUIREMENTS_FILE"
echo "Installing accelerate config"
mkdir -p "$HOME/.cache/huggingface/accelerate" mkdir -p "$HOME/.cache/huggingface/accelerate"
cp "${DIR}/config_files/accelerate/default_config.yaml"\ cp "${DIR}/config_files/accelerate/default_config.yaml"\
"$HOME/.cache/huggingface/accelerate/default_config.yaml" "$HOME/.cache/huggingface/accelerate/default_config.yaml"
@ -45,5 +50,14 @@ pushd /workspace
curl -L -O -J https://civitai.com/api/download/models/29792 curl -L -O -J https://civitai.com/api/download/models/29792
popd popd
echo "Creating directories"
mkdir -p /workspace/images/train/img \
/workspace/images/train/model \
/workspace/images/train/log
echo "Cleanup"
rm -r ~/.cache/huggingface ~/.cache/pip*
print "Setup complete." print "Setup complete."
exit 0 exit 0