let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

auto switch mps device to cpu device

Browse Source
This commit is contained in:
Qing
2023-02-11 13:30:09 +08:00
parent f9b5dcbfd7
commit 8f8bcfe0f4
15 changed files with 52 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lama_cleaner/web_config.py
View File

@@ -22,7 +22,8 @@ from lama_cleaner.const import (
DEFAULT_MODEL,
DEFAULT_DEVICE,
NO_GUI_AUTO_CLOSE_HELP,
DEFAULT_MODEL_DIR, MPS_SUPPORT_MODELS,
DEFAULT_MODEL_DIR,
MPS_SUPPORT_MODELS,
)
_config_file = None
@@ -115,7 +116,7 @@ def main(config_file: str):
with gr.Row():
model = gr.Radio(AVAILABLE_MODELS, label="Model", value=init_config.model)
device = gr.Radio(
AVAILABLE_DEVICES, label=f"Device(mps supports {MPS_SUPPORT_MODELS})", value=init_config.device
AVAILABLE_DEVICES, label="Device", value=init_config.device
)
gui = gr.Checkbox(init_config.gui, label=f"{GUI_HELP}")
no_gui_auto_close = gr.Checkbox(