Updates for runpod

gui.sh

@@ -29,9 +29,10 @@ SCRIPT_DIR=$(cd -- "$(dirname -- "$0")" && pwd)
 # Step into GUI local directory
 cd "$SCRIPT_DIR" || exit 1
 
-if [ "$RUNPOD" = false ]; then
-    # Activate the virtual environment
+if [ -d "$SCRIPT_DIR/venv" ]; then
     source "$SCRIPT_DIR/venv/bin/activate" || exit 1
+else
+    echo "venv folder does not exist. Not activating..."
 fi
 
 # Check if LD_LIBRARY_PATH environment variable exists

requirements_runpod.txt

@@ -1,29 +1,5 @@
-torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage
 xformers==0.0.20 bitsandbytes==0.39.1
-accelerate==0.19.0 tensorboard==2.12.1 tensorflow==2.12.0
+accelerate==0.15.0 tensorboard==2.12.1 tensorflow==2.12.0
 tensorrt
-albumentations==1.3.0
-altair==5.0.1
-dadaptation==3.1
-diffusers[torch]==0.10.2
-easygui==0.98.3
-einops==0.4.1
-fairscale==0.4.13
-ftfy==6.1.1
-gradio==3.32.0
-huggingface-hub>=0.13.3
-lion-pytorch==0.0.6
-lycoris_lora==0.1.6
-opencv-python==4.7.0.72
-prodigyopt==1.0
-pytorch-lightning==1.9.4
-rich==13.4.2
-safetensors==0.2.6
-timm==0.6.12
-tk==0.1.0
-toml==0.10.2
-transformers==4.29.2
-voluptuous==0.13.1
-wandb==0.15.0
-# for kohya_ss library
--e . # no_verify leave this to specify not checking this a verification stage
+-r requirements.txt