Update downs.sh

downs.sh

@@ -7,17 +7,24 @@ CHECKPOINT=$1
 RC=$2
 
 GOORM="0"
+RUNPOD="0"
+
 if [ -d "/workspace/firstContainer" ]; then
   GOORM="1"
+elif [ -d "/workspace" ]; then
+  RUNPOD="1"
 fi
 
 RELATIVE_DIR=`dirname "$0"`
 cd $RELATIVE_DIR
 
-if [ ! -d "SD" ]; then
+if [ -d "SD" ]; then
   cd SD
+elif [ -d "CU" ]; then
+  cd CU
 else
   cd SD*
+  cd CU*
 fi
 
 cd models
@@ -59,9 +66,15 @@ case $CHECKPOINT in
     d)
         CK_LINK="https://huggingface.co/zuv0/test/resolve/main/noobaiXLNAIXL_earlyAccessVersion.safetensors"
         ;;
+    d2)
+        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/noobaiXLNAIXL_epsilonPred05Version.safetensors"
+        ;;
     e)
         CK_LINK="https://huggingface.co/hanzogak/comradeshipXL/resolve/main/comradeshipXL-v14T13.safetensors"
         ;;
+    f)
+        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/illustriousXPonyMix_v3.safetensors"
+        ;;
     gc)
 		rm -rf *.safetensors
 		echo "CHECKPOINT remove."