Update color of command for print

wkpark · Mar 8, 2023 · cd0d72c · cd0d72c
1 parent 3a5d491
commit cd0d72c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lora_gui.py b/lora_gui.py
@@ -575,8 +575,8 @@ def train_model(
     )
 
     if  print_only_bool:
-        print('Here is the trainer command as a reference. It will not be executed:')
-        print(run_cmd)
+        print('\033[93m\nHere is the trainer command as a reference. It will not be executed:\033[0m\n')
+        print('\033[96m' + run_cmd + '\033[0m\n')
     else:
         print(run_cmd)
         # Run the command