Fix finetuner gui code issues

bmaltais · Jul 24, 2023 · fcf087c · fcf087c
1 parent 4849ea8
commit fcf087c
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/finetune_gui.py b/finetune_gui.py
@@ -423,8 +423,13 @@ def train_model(
             run_cmd += f' --full_path'
 
         log.info(run_cmd)
-
-        executor.execute_command(run_cmd=run_cmd)
+
+        if not print_only_bool:
+            # Run the command
+            if os.name == 'posix':
+                os.system(run_cmd)
+            else:
+                subprocess.run(run_cmd)
 
     # create images buckets
     if generate_image_buckets:
@@ -447,7 +452,10 @@ def train_model(
 
         if not print_only_bool:
             # Run the command
-            executor.execute_command(run_cmd=run_cmd)
+            if os.name == 'posix':
+                os.system(run_cmd)
+            else:
+                subprocess.run(run_cmd)
 
     image_num = len(
         [

diff --git a/library/class_basic_training.py b/library/class_basic_training.py
@@ -107,6 +107,8 @@ def __init__(
                     'DAdaptSGD',
                     'Lion',
                     'Lion8bit',
+                    "PagedAdamW8bit",
+                    "PagedLion8bit",
                     'Prodigy',
                     'SGDNesterov',
                     'SGDNesterov8bit',

diff --git a/requirements_windows_torch2.txt b/requirements_windows_torch2.txt
@@ -1,4 +1,5 @@
 torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118 # no_verify
-xformers==0.0.20 bitsandbytes==0.35.0
+xformers==0.0.20
+https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.39.1-py3-none-win_amd64.whl # no_verify
 tensorboard==2.12.3 tensorflow==2.12.0
 -r requirements.txt