better api defaults

This commit is contained in:
Vladimir Mandic
2024-08-28 10:58:54 -04:00
parent 65c137abf1
commit bb7f84b99b
3 changed files with 8 additions and 4 deletions
+3 -3
View File
@@ -201,7 +201,7 @@ ReqTxt2Img = PydanticModelGenerator(
"StableDiffusionProcessingTxt2Img",
StableDiffusionProcessingTxt2Img,
[
{"key": "sampler_index", "type": str, "default": "Euler"},
{"key": "sampler_index", "type": str, "default": "UniPC"},
{"key": "script_name", "type": str, "default": None},
{"key": "script_args", "type": list, "default": []},
{"key": "send_images", "type": bool, "default": True},
@@ -222,9 +222,9 @@ ReqImg2Img = PydanticModelGenerator(
"StableDiffusionProcessingImg2Img",
StableDiffusionProcessingImg2Img,
[
{"key": "sampler_index", "type": str, "default": "Euler"},
{"key": "sampler_index", "type": str, "default": "UniPC"},
{"key": "init_images", "type": list, "default": None},
{"key": "denoising_strength", "type": float, "default": 0.75},
{"key": "denoising_strength", "type": float, "default": 0.5},
{"key": "mask", "type": str, "default": None},
{"key": "include_init_images", "type": bool, "default": False, "exclude": True},
{"key": "script_name", "type": str, "default": None},
+4 -1
View File
@@ -538,7 +538,10 @@ def save_intermediate(p, latents, suffix):
def update_sampler(p, sd_model, second_pass=False):
sampler_selection = p.hr_sampler_name if second_pass else p.sampler_name
if hasattr(sd_model, 'scheduler'):
sampler = sd_samplers.all_samplers_map.get(sampler_selection, None)
if sampler_selection is None or sampler_selection == 'None':
sampler = sd_samplers.all_samplers_map.get("UniPC")
else:
sampler = sd_samplers.all_samplers_map.get(sampler_selection, None)
if sampler is None:
shared.log.warning(f'Sampler: sampler="{sampler_selection}" not found')
sampler = sd_samplers.all_samplers_map.get("UniPC")
+1
View File
@@ -140,6 +140,7 @@ class FaceRestorerYolo(FaceRestoration):
}
if args['denoising_strength'] == 0:
shared.log.debug('Face HiRes skip: strength=0')
return np_image
control_pipeline = None
orig_class = shared.sd_model.__class__
if getattr(p, 'is_control', False):