kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

always cpu_textencoder=False if device is not cuda

Browse Source
This commit is contained in:
Qing 2024-01-10 13:34:51 +08:00
parent 38b6edacf0
commit 772ef65f7b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
iopaint/cli.py
View File

@ -106,7 +106,9 @@ def start(
file_okay=False, file_okay=False,
callback=setup_model_dir, callback=setup_model_dir,
), ),
low_mem: bool = Option(False, help="Enable attention slicing and vae tiling to save memory."), low_mem: bool = Option(
False, help="Enable attention slicing and vae tiling to save memory."
),
no_half: bool = Option(False, help=NO_HALF_HELP), no_half: bool = Option(False, help=NO_HALF_HELP),
cpu_offload: bool = Option(False, help=CPU_OFFLOAD_HELP), cpu_offload: bool = Option(False, help=CPU_OFFLOAD_HELP),
disable_nsfw_checker: bool = Option(False, help=DISABLE_NSFW_HELP), disable_nsfw_checker: bool = Option(False, help=DISABLE_NSFW_HELP),
@ -174,7 +176,7 @@ def start(
low_mem=low_mem, low_mem=low_mem,
cpu_offload=cpu_offload, cpu_offload=cpu_offload,
disable_nsfw_checker=disable_nsfw_checker, disable_nsfw_checker=disable_nsfw_checker,
cpu_textencoder=cpu_textencoder, cpu_textencoder=cpu_textencoder if device == Device.cuda else False,
device=device, device=device,
gui=gui, gui=gui,
disable_model_switch=disable_model_switch, disable_model_switch=disable_model_switch,