From 14dc16ccd4f00d5b5d91ec4b5c797986fbaa6788 Mon Sep 17 00:00:00 2001
From: JSTayco <jstayco@protonmail.ch>
Date: Sat, 1 Apr 2023 10:18:44 -0700
Subject: [PATCH] Minor fix to branch name

---
 setup.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.sh b/setup.sh
index 7b40b49..80877aa 100755
--- 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