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

4
iopaint/model/sd.py
View File

@@ -29,7 +29,7 @@ class SD(DiffusionInpaintModel):
**kwargs.get("pipe_components", {}),
"local_files_only": is_local_files_only(**kwargs),
}
disable_nsfw_checker = kwargs["disable_nsfw"] or kwargs.get(
disable_nsfw_checker = kwargs.get("disable_nsfw", False) or kwargs.get(
"cpu_offload", False
)
if disable_nsfw_checker:
@@ -71,7 +71,7 @@ class SD(DiffusionInpaintModel):
self.model.enable_sequential_cpu_offload(gpu_id=0)
else:
self.model = self.model.to(device)
if kwargs["sd_cpu_textencoder"]:
if kwargs.get("sd_cpu_textencoder", False):
logger.info("Run Stable Diffusion TextEncoder on CPU")
self.model.text_encoder = CPUTextEncoderWrapper(
self.model.text_encoder, torch_dtype