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

update

Browse Source
This commit is contained in:
Qing 2024-07-04 08:58:07 +08:00
parent ef1f7a5324
commit 9afdbd1c0a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
iopaint/batch_processing.py
View File

@ -35,7 +35,6 @@ def glob_images(path: Path) -> Dict[str, Path]:
return res return res
def batch_inpaint( def batch_inpaint(
model: str, model: str,
device, device,
@ -67,6 +66,7 @@ def batch_inpaint(
else: else:
with open(config, "r", encoding="utf-8") as f: with open(config, "r", encoding="utf-8") as f:
inpaint_request = InpaintRequest(**json.load(f)) inpaint_request = InpaintRequest(**json.load(f))
logger.info(f"Using config: {inpaint_request}")
model_manager = ModelManager(name=model, device=device) model_manager = ModelManager(name=model, device=device)
first_mask = list(mask_paths.values())[0] first_mask = list(mask_paths.values())[0]