One-click installer: minor change to CMD_FLAGS.txt in CPU mode

This commit is contained in:
oobabooga 2024-03-03 17:42:59 -08:00
parent 8bd4960d05
commit fa4ce0eee8

View File

@ -239,7 +239,7 @@ def install_webui():
with open(cmd_flags_path, 'r+') as cmd_flags_file: with open(cmd_flags_path, 'r+') as cmd_flags_file:
if "--cpu" not in cmd_flags_file.read(): if "--cpu" not in cmd_flags_file.read():
print_big_message("Adding the --cpu flag to CMD_FLAGS.txt.") 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 # Check if the user wants CUDA 11.8
elif any((is_windows(), is_linux())) and selected_gpu == "NVIDIA": elif any((is_windows(), is_linux())) and selected_gpu == "NVIDIA":