From a97ae99ddb14808fbb0bca00bfa52842ea99307c Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Wed, 19 Jun 2024 02:45:29 +0800
Subject: [PATCH] 'update'

---
 mz_train_tools_core.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mz_train_tools_core.py b/mz_train_tools_core.py
index 18d2ceb..f7c1975 100644
--- a/mz_train_tools_core.py
+++ b/mz_train_tools_core.py
@@ -33,9 +33,9 @@ def MZ_KohyaSSInitWorkspace_call(args={}):
         result = subprocess.run(
             ["git", "branch"], cwd=kohya_ss_lora_dir, stdout=subprocess.PIPE, check=True)
 
-        print(f"current branch: {result.stdout}")
+        print(f"current branch: {result.stdout.decode()}")
 
-        if branch.encode() not in result.stdout:
+        if branch not in result.stdout.decode():
             subprocess.run(
                 ["git", "remote", "set-branches", "origin", branch], cwd=kohya_ss_lora_dir, check=True)
             subprocess.run(