diff --git a/stable-diffusion/setup_runpod_lora_learn.sh b/stable-diffusion/setup_runpod_lora_learn.sh index 6f713cf..6a375af 100644 --- a/stable-diffusion/setup_runpod_lora_learn.sh +++ b/stable-diffusion/setup_runpod_lora_learn.sh @@ -33,7 +33,7 @@ echo "Installing the project..." DIR="/workspace/kohya_ss" 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_unix.txt" >"$TEMP_REQUIREMENTS_FILE" python -m pip install --use-pep517 --upgrade -r "$TEMP_REQUIREMENTS_FILE" echo "Installing accelerate config" @@ -47,7 +47,7 @@ VENV_DIR="${VENV_DIR%/lib/python3.10/site-packages}" echo "Downloading base model" pushd /workspace -curl -L -O -J https://civitai.com/api/download/models/29792 +curl -L -o model.safetensors -J https://civitai.com/api/download/models/28562 popd echo "Creating directories"