add remove bg model selection

2024-02-08 16:49:54 +08:00
parent cf9ceea4e6
commit 8060e16c70
19 changed files with 915 additions and 222 deletions
--- a/iopaint/model_manager.py
+++ b/iopaint/model_manager.py
@@ -8,8 +8,7 @@ from iopaint.download import scan_models
 from iopaint.helper import switch_mps_device
 from iopaint.model import models, ControlNet, SD, SDXL
 from iopaint.model.utils import torch_gc, is_local_files_only
-from iopaint.model_info import ModelInfo, ModelType
-from iopaint.schema import InpaintRequest
+from iopaint.schema import InpaintRequest, ModelInfo, ModelType


 class ModelManager: