


"""simple docstring"""






from ...utils import (
    OptionalDependencyNotAvailable,
    is_torch_available,
    is_transformers_available,
    is_transformers_version,
)


try:
					if not (is_transformers_available() and is_torch_available()):
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					from ...utils.dummy_torch_and_transformers_objects import (
					    ImageTextPipelineOutput,
					    UniDiffuserPipeline,
					)
else:
					from .modeling_text_decoder import UniDiffuserTextDecoder
					from .modeling_uvit import UniDiffuserModel, UTransformeraDModel
					from .pipeline_unidiffuser import ImageTextPipelineOutput, UniDiffuserPipeline





"""simple docstring"""






UpperCamelCase__				:Tuple			   =	frozenset(
    [
        """prompt""",
        """height""",
        """width""",
        """guidance_scale""",
        """negative_prompt""",
        """prompt_embeds""",
        """negative_prompt_embeds""",
        """cross_attention_kwargs""",
    ]
)

UpperCamelCase__				:List[str]			   =	frozenset(["""prompt""", """negative_prompt"""])

UpperCamelCase__				:Tuple			   =	frozenset([])

UpperCamelCase__				:Union[str, Any]			   =	frozenset(["""image"""])

UpperCamelCase__				:str			   =	frozenset(
    [
        """image""",
        """height""",
        """width""",
        """guidance_scale""",
    ]
)

UpperCamelCase__				:Any			   =	frozenset(["""image"""])

UpperCamelCase__				:Optional[int]			   =	frozenset(
    [
        """prompt""",
        """image""",
        """height""",
        """width""",
        """guidance_scale""",
        """negative_prompt""",
        """prompt_embeds""",
        """negative_prompt_embeds""",
    ]
)

UpperCamelCase__				:int			   =	frozenset(["""prompt""", """image""", """negative_prompt"""])

UpperCamelCase__				:Tuple			   =	frozenset(
    [
        # Text guided image variation with an image mask
        """prompt""",
        """image""",
        """mask_image""",
        """height""",
        """width""",
        """guidance_scale""",
        """negative_prompt""",
        """prompt_embeds""",
        """negative_prompt_embeds""",
    ]
)

UpperCamelCase__				:Any			   =	frozenset(["""prompt""", """image""", """mask_image""", """negative_prompt"""])

UpperCamelCase__				:Dict			   =	frozenset(
    [
        # image variation with an image mask
        """image""",
        """mask_image""",
        """height""",
        """width""",
        """guidance_scale""",
    ]
)

UpperCamelCase__				:Dict			   =	frozenset(["""image""", """mask_image"""])

UpperCamelCase__				:Tuple			   =	frozenset(
    [
        """example_image""",
        """image""",
        """mask_image""",
        """height""",
        """width""",
        """guidance_scale""",
    ]
)

UpperCamelCase__				:List[str]			   =	frozenset(["""example_image""", """image""", """mask_image"""])

UpperCamelCase__				:List[str]			   =	frozenset(["""class_labels"""])

UpperCamelCase__				:Union[str, Any]			   =	frozenset(["""class_labels"""])

UpperCamelCase__				:Any			   =	frozenset(["""batch_size"""])

UpperCamelCase__				:Optional[Any]			   =	frozenset([])

UpperCamelCase__				:Optional[Any]			   =	frozenset(["""batch_size"""])

UpperCamelCase__				:Optional[int]			   =	frozenset([])

UpperCamelCase__				:str			   =	frozenset(
    [
        """prompt""",
        """audio_length_in_s""",
        """guidance_scale""",
        """negative_prompt""",
        """prompt_embeds""",
        """negative_prompt_embeds""",
        """cross_attention_kwargs""",
    ]
)

UpperCamelCase__				:Union[str, Any]			   =	frozenset(["""prompt""", """negative_prompt"""])
UpperCamelCase__				:Tuple			   =	frozenset(["""input_tokens"""])

UpperCamelCase__				:int			   =	frozenset(["""input_tokens"""])

