use subprocess.run(run_cmd, shell=True, env=env)

kohya_gui/basic_caption_gui.py

@@ -3,12 +3,14 @@ from easygui import msgbox
 import subprocess
 from .common_gui import get_folder_path, add_pre_postfix, find_replace, scriptdir
 import os
+import sys
 
 from .custom_logging import setup_logging
 
 # Set up logging
 log = setup_logging()
 
+PYTHON = sys.executable
 
 def caption_images(
     caption_text,
@@ -36,7 +38,7 @@ def caption_images(
         log.info(f'Captioning files in {images_dir} with {caption_text}...')
 
         # Build the command to run caption.py
-        run_cmd = fr'python "{scriptdir}/tools/caption.py"'
+        run_cmd = fr'{PYTHON} "{scriptdir}/tools/caption.py"'
         run_cmd += f' --caption_text="{caption_text}"'
 
         # Add optional flags to the command
@@ -49,11 +51,11 @@ def caption_images(
 
         log.info(run_cmd)
 
+        env = os.environ.copy()
+        env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
         # Run the command based on the operating system
-        if os.name == 'posix':
-            os.system(run_cmd)
-        else:
-            subprocess.run(run_cmd)
+        subprocess.run(run_cmd, shell=True, env=env)
 
     # Check if overwrite option is enabled
     if overwrite:

kohya_gui/blip_caption_gui.py

@@ -52,11 +52,11 @@ def caption_images(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
     # Add prefix and postfix
     add_pre_postfix(

kohya_gui/convert_lcm_gui.py

@@ -39,11 +39,11 @@ def convert_lcm(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == "posix":
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
     # Return a success message
     log.info("Done extracting...")

kohya_gui/convert_model_gui.py

@@ -103,11 +103,11 @@ def convert_model(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
     if (
         not target_model_type == 'diffuser'

kohya_gui/extract_lora_from_dylora_gui.py

@@ -46,11 +46,11 @@ def extract_dylora(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
     log.info('Done extracting DyLoRA...')
 

kohya_gui/extract_lora_gui.py

@@ -84,11 +84,11 @@ def extract_lora(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
 
 ###

kohya_gui/extract_lycoris_locon_gui.py

@@ -92,11 +92,11 @@ def extract_lycoris_locon(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == "posix":
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
         
     log.info('Done extracting...')
 

kohya_gui/git_caption_gui.py

@@ -47,11 +47,11 @@ def caption_images(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
     # Add prefix and postfix
     add_pre_postfix(

kohya_gui/group_images_gui.py

@@ -47,10 +47,11 @@ def group_images(
 
     log.info(run_cmd)
 
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
+    # Run the command
+    subprocess.run(run_cmd, shell=True, env=env)
 
     log.info('...grouping done')
 

kohya_gui/merge_lora_gui.py

@@ -312,10 +312,10 @@ class GradioMergeLoRaTab:
 
         log.info(run_cmd)
 
+        env = os.environ.copy()
+        env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
         # Run the command
-        if os.name == 'posix':
-            os.system(run_cmd)
-        else:
-            subprocess.run(run_cmd)
+        subprocess.run(run_cmd, shell=True, env=env)
 
         log.info('Done merging...')

kohya_gui/merge_lycoris_gui.py

@@ -48,11 +48,11 @@ def merge_lycoris(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
     log.info('Done merging...')
 

kohya_gui/resize_lora_gui.py

@@ -73,11 +73,11 @@ def resize_lora(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
     log.info('Done resizing...')
 

kohya_gui/svd_merge_lora_gui.py

@@ -92,11 +92,11 @@ def svd_merge_lora(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
 
 ###

kohya_gui/verify_lora_gui.py

@@ -44,9 +44,12 @@ def verify_lora(
 
     log.info(' '.join(run_cmd))
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
     process = subprocess.Popen(
-        run_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE
+        run_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=env,
     )
     output, error = process.communicate()
 

kohya_gui/wd14_caption_gui.py

@@ -72,11 +72,11 @@ def caption_images(
 
     log.info(run_cmd)
 
+    env = os.environ.copy()
+    env['PYTHONPATH'] = fr"{scriptdir}{os.pathsep}{env.get('PYTHONPATH', '')}"
+
     # Run the command
-    if os.name == 'posix':
-        os.system(run_cmd)
-    else:
-        subprocess.run(run_cmd)
+    subprocess.run(run_cmd, shell=True, env=env)
 
     # Add prefix and postfix
     add_pre_postfix(