1
0
Fork 0

fix(api): use min/max from config for more params

This commit is contained in:
Sean Sube 2023-04-01 18:14:18 -05:00
parent 85b332467e
commit 56ff9022da
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 83 additions and 15 deletions

View File

@ -86,14 +86,18 @@ def expand_prompt(
if skip_clip_states > 0:
layer_norm = torch.nn.LayerNorm(last_state.shape[2])
norm_state = layer_norm(
torch.from_numpy(hidden_states[-skip_clip_states].astype(np.float32)).detach()
torch.from_numpy(
hidden_states[-skip_clip_states].astype(np.float32)
).detach()
)
logger.trace(
"normalized results after skipping %s layers: %s",
skip_clip_states,
norm_state.shape,
)
group_embeds.append(norm_state)
group_embeds.append(
norm_state.numpy().astype(hidden_states[-skip_clip_states].dtype)
)
else:
group_embeds.append(last_state)

View File

@ -139,30 +139,76 @@ def pipeline_from_request(
def border_from_request() -> Border:
left = get_and_clamp_int(
request.args, "left", 0, get_config_value("width", "max"), 0
request.args,
"left",
get_config_value("width"),
get_config_value("width", "max"),
get_config_value("width", "min"),
)
right = get_and_clamp_int(
request.args, "right", 0, get_config_value("width", "max"), 0
request.args,
"right",
get_config_value("width"),
get_config_value("width", "max"),
get_config_value("width", "min"),
)
top = get_and_clamp_int(
request.args, "top", 0, get_config_value("height", "max"), 0
request.args,
"top",
get_config_value("width"),
get_config_value("height", "max"),
get_config_value("height", "min"),
)
bottom = get_and_clamp_int(
request.args, "bottom", 0, get_config_value("height", "max"), 0
request.args,
"bottom",
get_config_value("width"),
get_config_value("height", "max"),
get_config_value("height", "min"),
)
return Border(left, right, top, bottom)
def upscale_from_request() -> UpscaleParams:
denoise = get_and_clamp_float(request.args, "denoise", 0.5, 1.0, 0.0)
scale = get_and_clamp_int(request.args, "scale", 1, 4, 1)
outscale = get_and_clamp_int(request.args, "outscale", 1, 4, 1)
denoise = get_and_clamp_float(
request.args,
"denoise",
get_config_value("denoise"),
get_config_value("denoise", "max"),
get_config_value("denoise", "min"),
)
scale = get_and_clamp_int(
request.args,
"scale",
get_config_value("scale"),
get_config_value("scale", "max"),
get_config_value("scale", "min"),
)
outscale = get_and_clamp_int(
request.args,
"outscale",
get_config_value("outscale"),
get_config_value("outscale", "max"),
get_config_value("outscale", "min"),
)
upscaling = get_from_list(request.args, "upscaling", get_upscaling_models())
correction = get_from_list(request.args, "correction", get_correction_models())
faces = get_not_empty(request.args, "faces", "false") == "true"
face_outscale = get_and_clamp_int(request.args, "faceOutscale", 1, 4, 1)
face_strength = get_and_clamp_float(request.args, "faceStrength", 0.5, 1.0, 0.0)
face_outscale = get_and_clamp_int(
request.args,
"faceOutscale",
get_config_value("faceOutscale"),
get_config_value("faceOutscale", "max"),
get_config_value("faceOutscale", "min"),
)
face_strength = get_and_clamp_float(
request.args,
"faceStrength",
get_config_value("faceStrength"),
get_config_value("faceStrength", "max"),
get_config_value("faceStrength", "min"),
)
upscale_order = request.args.get("upscaleOrder", "correction-first")
return UpscaleParams(
@ -181,9 +227,27 @@ def upscale_from_request() -> UpscaleParams:
def highres_from_request() -> HighresParams:
method = get_from_list(request.args, "highresMethod", get_highres_methods())
scale = get_and_clamp_int(request.args, "highresScale", 1, 4, 1)
steps = get_and_clamp_int(request.args, "highresSteps", 1, 200, 1)
strength = get_and_clamp_float(request.args, "highresStrength", 0.5, 1.0, 0.0)
scale = get_and_clamp_int(
request.args,
"highresScale",
get_config_value("highresScale"),
get_config_value("highresScale", "max"),
get_config_value("highresScale", "min"),
)
steps = get_and_clamp_int(
request.args,
"highresSteps",
get_config_value("highresSteps"),
get_config_value("highresSteps", "max"),
get_config_value("highresSteps", "min"),
)
strength = get_and_clamp_float(
request.args,
"highresStrength",
get_config_value("highresStrength"),
get_config_value("highresStrength"),
get_config_value("highresStrength", "min"),
)
return HighresParams(
scale,

View File

@ -81,7 +81,7 @@
"step": 1
},
"highresStrength": {
"default": 0.5,
"default": 0.1,
"min": 0,
"max": 1,
"step": 0.01