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

set controlnet_conditioning_scale=0 when use extender

Browse Source
This commit is contained in:
Qing 2024-01-10 21:25:34 +08:00
parent 7545d54870
commit dfcb9017f1
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
iopaint/schema.py
View File

@ -3,6 +3,7 @@ from enum import Enum
from pathlib import Path
from typing import Optional, Literal, List
from loguru import logger
from pydantic import BaseModel, Field, field_validator
from iopaint.const import Device, InteractiveSegModel, RealESRGANModel
@ -227,7 +228,7 @@ class InpaintRequest(BaseModel):
# ControlNet
enable_controlnet: bool = Field(False, description="Enable controlnet")
controlnet_conditioning_scale: float = Field(
0.4, description="Conditioning scale", gt=0.0, le=1.0
0.4, description="Conditioning scale", ge=0.0, le=1.0
)
controlnet_method: str = Field(
"lllyasviel/control_v11p_sd15_canny", description="Controlnet method"
@ -251,6 +252,16 @@ class InpaintRequest(BaseModel):
return random.randint(1, 99999999)
return v
@field_validator("controlnet_conditioning_scale")
@classmethod
def validate_field(cls, v: float, values):
use_extender = values.data["use_extender"]
enable_controlnet = values.data["enable_controlnet"]
if use_extender and enable_controlnet:
logger.info(f"Extender is enabled, set controlnet_conditioning_scale=0")
return 0
return v
class RunPluginRequest(BaseModel):
name: str
@ -296,6 +307,8 @@ AdjustMaskOperate = Literal["expand", "shrink"]
class AdjustMaskRequest(BaseModel):
mask: str = Field(..., description="base64 encoded mask. 255 means area to do inpaint")
mask: str = Field(
..., description="base64 encoded mask. 255 means area to do inpaint"
)
operate: AdjustMaskOperate = Field(..., description="expand or shrink")
kernel_size: int = Field(5, description="Kernel size for expanding mask")