Minor fix to branch name

gobomus · Apr 1, 2023 · 14dc16c · 14dc16c
1 parent 69deea0
commit 14dc16c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.sh b/setup.sh
@@ -357,7 +357,7 @@ update_kohya_ss() {
         fi
 
         if [ "$MANUAL_BRANCH_SWITCH" = false ]; then
-          git -C "$DIR" switch "$BRANCH" >&3
+          git -C "$DIR" switch "$(git rev-parse --abbrev-ref HEAD)" >&3
         else
           if ! git -C "$DIR" switch "$BRANCH" >&4; then
             echo "Branch $BRANCH did not exist. Creating it." >&4