diff --git a/colabkobold.sh b/colabkobold.sh index 3adfbc9d..1764a166 100644 --- a/colabkobold.sh +++ b/colabkobold.sh @@ -2,7 +2,7 @@ # KoboldAI Easy Colab Deployment Script by Henk717 # read the options -TEMP=`getopt -o m:i:p:c:d:x:a:l:z:g:t:n:b:s:o: --long model:,init:,path:,configname:,download:,aria2:,dloc:xloc:7z:git:tar:ngrok:branch:savemodel:localtunnel: -- "$@"` +TEMP=`getopt -o m:i:p:c:d:x:a:l:z:g:t:n:b:s: --long model:,init:,path:,configname:,download:,aria2:,dloc:,xloc:,7z:,git:,tar:,ngrok:,branch:,savemodel:,localtunnel:,lt: -- "$@"` eval set -- "$TEMP" # extract options and their arguments into variables. @@ -18,7 +18,7 @@ while true ; do configname=" --configname $2" ; shift 2 ;; -n|--ngrok) ngrok=" --ngrok" ; shift 2 ;; - -o|--localtunnel) + --lt|--localtunnel) localtunnel=" --localtunnel" ; shift 2 ;; -d|--download) download="$2" ; shift 2 ;; @@ -52,7 +52,7 @@ function launch exit 0 else cd /content/KoboldAI-Client - echo "Launching KoboldAI with the following options : python3 aiserver.py$model$kmpath$configname$ngrok$savemodel --colab" + echo "Launching KoboldAI with the following options : python3 aiserver.py$model$kmpath$configname$ngrok$localtunnel$savemodel --colab" python3 aiserver.py$model$kmpath$configname$ngrok$localtunnel$savemodel --colab exit fi