feat: add API parameter for upscale checkbox
This commit is contained in:
parent
a5ecb59cd9
commit
71fbc871a6
|
@ -67,7 +67,9 @@ def stage_upscale_correction(
|
|||
**kwargs,
|
||||
"upscale": upscale,
|
||||
}
|
||||
|
||||
upscale_stage: Optional[PipelineStage] = None
|
||||
if upscale.upscale:
|
||||
if "bsrgan" in upscale.upscale_model:
|
||||
bsrgan_params = StageParams(
|
||||
tile_size=stage.tile_size,
|
||||
|
|
|
@ -391,6 +391,7 @@ class UpscaleParams:
|
|||
upscale_model: str,
|
||||
correction_model: Optional[str] = None,
|
||||
denoise: float = 0.5,
|
||||
upscale=True,
|
||||
faces=True,
|
||||
face_outscale: int = 1,
|
||||
face_strength: float = 0.5,
|
||||
|
@ -406,6 +407,7 @@ class UpscaleParams:
|
|||
self.upscale_model = upscale_model
|
||||
self.correction_model = correction_model
|
||||
self.denoise = denoise
|
||||
self.upscale = upscale
|
||||
self.faces = faces
|
||||
self.face_outscale = face_outscale
|
||||
self.face_strength = face_strength
|
||||
|
@ -421,6 +423,7 @@ class UpscaleParams:
|
|||
self.upscale_model,
|
||||
correction_model=self.correction_model,
|
||||
denoise=self.denoise,
|
||||
upscale=self.upscale,
|
||||
faces=self.faces,
|
||||
face_outscale=self.face_outscale,
|
||||
face_strength=self.face_strength,
|
||||
|
@ -447,6 +450,7 @@ class UpscaleParams:
|
|||
"upscale_model": self.upscale_model,
|
||||
"correction_model": self.correction_model,
|
||||
"denoise": self.denoise,
|
||||
"upscale": self.upscale,
|
||||
"faces": self.faces,
|
||||
"face_outscale": self.face_outscale,
|
||||
"face_strength": self.face_strength,
|
||||
|
@ -463,6 +467,7 @@ class UpscaleParams:
|
|||
kwargs.get("upscale_model", self.upscale_model),
|
||||
kwargs.get("correction_model", self.correction_model),
|
||||
kwargs.get("denoise", self.denoise),
|
||||
kwargs.get("upscale", self.upscale),
|
||||
kwargs.get("faces", self.faces),
|
||||
kwargs.get("face_outscale", self.face_outscale),
|
||||
kwargs.get("face_strength", self.face_strength),
|
||||
|
|
|
@ -239,6 +239,7 @@ def build_upscale(
|
|||
if data is None:
|
||||
data = request.args
|
||||
|
||||
upscale = get_boolean(data, "upscale", False)
|
||||
denoise = get_and_clamp_float(
|
||||
data,
|
||||
"denoise",
|
||||
|
@ -262,7 +263,8 @@ def build_upscale(
|
|||
)
|
||||
upscaling = get_from_list(data, "upscaling", get_upscaling_models())
|
||||
correction = get_from_list(data, "correction", get_correction_models())
|
||||
faces = get_not_empty(data, "faces", "false") == "true"
|
||||
|
||||
faces = get_boolean(data, "faces", False)
|
||||
face_outscale = get_and_clamp_int(
|
||||
data,
|
||||
"faceOutscale",
|
||||
|
@ -283,6 +285,7 @@ def build_upscale(
|
|||
upscaling,
|
||||
correction_model=correction,
|
||||
denoise=denoise,
|
||||
upscale=upscale,
|
||||
faces=faces,
|
||||
face_outscale=face_outscale,
|
||||
face_strength=face_strength,
|
||||
|
|
|
@ -110,6 +110,7 @@ export function appendModelToURL(url: URL, params: ModelParams) {
|
|||
* Append the upscale parameters to an existing URL.
|
||||
*/
|
||||
export function appendUpscaleToURL(url: URL, upscale: UpscaleParams) {
|
||||
url.searchParams.append('upscale', String(upscale.enabled));
|
||||
url.searchParams.append('upscaleOrder', upscale.upscaleOrder);
|
||||
|
||||
if (upscale.enabled) {
|
||||
|
|
Loading…
Reference in New Issue