From fa4ce0eee82c42a25629fdf1ff56863b72c5a912 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 3 Mar 2024 17:42:59 -0800
Subject: [PATCH] One-click installer: minor change to CMD_FLAGS.txt in CPU
 mode

---
 one_click.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/one_click.py b/one_click.py
index 4aa5595c..92663df4 100644
--- a/one_click.py
+++ b/one_click.py
@@ -239,7 +239,7 @@ def install_webui():
         with open(cmd_flags_path, 'r+') as cmd_flags_file:
             if "--cpu" not in cmd_flags_file.read():
                 print_big_message("Adding the --cpu flag to CMD_FLAGS.txt.")
-                cmd_flags_file.write("\n--cpu")
+                cmd_flags_file.write("\n--cpu\n")
 
     # Check if the user wants CUDA 11.8
     elif any((is_windows(), is_linux())) and selected_gpu == "NVIDIA":