Unverified Commit 87b800e1 authored by Aryan's avatar Aryan Committed by GitHub
Browse files

[modular diffusers] Fix AutoGuidance validation (#12247)

fix
parent e58711e7
...@@ -82,15 +82,15 @@ class AutoGuidance(BaseGuidance): ...@@ -82,15 +82,15 @@ class AutoGuidance(BaseGuidance):
self.guidance_rescale = guidance_rescale self.guidance_rescale = guidance_rescale
self.use_original_formulation = use_original_formulation self.use_original_formulation = use_original_formulation
if auto_guidance_layers is None and auto_guidance_config is None: is_layer_or_config_provided = auto_guidance_layers is not None or auto_guidance_config is not None
is_layer_and_config_provided = auto_guidance_layers is not None and auto_guidance_config is not None
if not is_layer_or_config_provided:
raise ValueError( raise ValueError(
"Either `auto_guidance_layers` or `auto_guidance_config` must be provided to enable Skip Layer Guidance." "Either `auto_guidance_layers` or `auto_guidance_config` must be provided to enable AutoGuidance."
) )
if auto_guidance_layers is not None and auto_guidance_config is not None: if is_layer_and_config_provided:
raise ValueError("Only one of `auto_guidance_layers` or `auto_guidance_config` can be provided.") raise ValueError("Only one of `auto_guidance_layers` or `auto_guidance_config` can be provided.")
if (dropout is None and auto_guidance_layers is not None) or ( if auto_guidance_config is None and dropout is None:
dropout is not None and auto_guidance_layers is None
):
raise ValueError("`dropout` must be provided if `auto_guidance_layers` is provided.") raise ValueError("`dropout` must be provided if `auto_guidance_layers` is provided.")
if auto_guidance_layers is not None: if auto_guidance_layers is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment