feature: skip composition at sizes slightly larger than model is expecting

pull/411/head^2
Bryce 6 months ago committed by Bryce Drennan
parent 203747b14f
commit 5a636e45c5

@ -256,11 +256,13 @@ def _generate_single_image(
controlnets.append((controlnet, control_image_t))
if prompt.allow_compose_phase:
cutoff_size = get_model_default_image_size(prompt.model_architecture)
cutoff_size = (int(cutoff_size[0] * 1.30), int(cutoff_size[1] * 1.30))
compose_kwargs = {
"prompt": prompt,
"target_height": prompt.height,
"target_width": prompt.width,
"cutoff": get_model_default_image_size(prompt.model_architecture),
"cutoff": cutoff_size,
"dtype": dtype,
}

Loading…
Cancel
Save