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

support brushnet sdxl model

Browse Source
This commit is contained in:
Starsky Wong
2025-02-12 14:21:11 +08:00
parent 65b37727b5
commit 4b5f8b8995
8 changed files with 1740 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
iopaint/model_manager.py
View File

@@ -8,6 +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.brushnet.brushnet_wrapper import BrushNetWrapper
from iopaint.model.brushnet.brushnet_xl_wrapper import BrushNetXLWrapper
from iopaint.model.power_paint.power_paint_v2 import PowerPaintV2
from iopaint.model.utils import torch_gc, is_local_files_only
from iopaint.schema import InpaintRequest, ModelInfo, ModelType
@@ -63,7 +64,10 @@ class ModelManager:
return ControlNet(device, **kwargs)
if model_info.support_brushnet and self.enable_brushnet:
return BrushNetWrapper(device, **kwargs)
if model_info.model_type == ModelType.DIFFUSERS_SD:
return BrushNetWrapper(device, **kwargs)
elif model_info.model_type == ModelType.DIFFUSERS_SDXL:
return BrushNetXLWrapper(device, **kwargs)
if model_info.support_powerpaint_v2 and self.enable_powerpaint_v2:
return PowerPaintV2(device, **kwargs)