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 pathlib import Path
from typing import Optional, Literal, List from typing import Optional, Literal, List
from loguru import logger
from pydantic import BaseModel, Field, field_validator from pydantic import BaseModel, Field, field_validator
from iopaint.const import Device, InteractiveSegModel, RealESRGANModel from iopaint.const import Device, InteractiveSegModel, RealESRGANModel
@ -227,7 +228,7 @@ class InpaintRequest(BaseModel):
# ControlNet # ControlNet
enable_controlnet: bool = Field(False, description="Enable controlnet") enable_controlnet: bool = Field(False, description="Enable controlnet")
controlnet_conditioning_scale: float = Field( 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( controlnet_method: str = Field(
"lllyasviel/control_v11p_sd15_canny", description="Controlnet method" "lllyasviel/control_v11p_sd15_canny", description="Controlnet method"
@ -251,6 +252,16 @@ class InpaintRequest(BaseModel):
return random.randint(1, 99999999) return random.randint(1, 99999999)
return v 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): class RunPluginRequest(BaseModel):
name: str name: str
@ -296,6 +307,8 @@ AdjustMaskOperate = Literal["expand", "shrink"]
class AdjustMaskRequest(BaseModel): 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") operate: AdjustMaskOperate = Field(..., description="expand or shrink")
kernel_size: int = Field(5, description="Kernel size for expanding mask") kernel_size: int = Field(5, description="Kernel size for expanding mask")