1717
1818import inspect
1919from collections .abc import Callable
20- from typing import Any , List , Optional , Union
20+ from typing import Any , Dict , List , Optional , Tuple , Union
2121
2222import numpy as np
2323import PIL
@@ -1211,8 +1211,8 @@ def prepare_control_image(
12111211 @replace_example_docstring (EXAMPLE_DOC_STRING )
12121212 def __call__ (
12131213 self ,
1214- prompt : Optional [Union [str , list [str ]]] = None ,
1215- prompt_2 : Optional [Union [str , list [str ]]] = None ,
1214+ prompt : Optional [Union [str , List [str ]]] = None ,
1215+ prompt_2 : Optional [Union [str , List [str ]]] = None ,
12161216 image : Optional [Union [torch .Tensor , PIL .Image .Image ]] = None ,
12171217 mask_image : Optional [Union [torch .Tensor , PIL .Image .Image ]] = None ,
12181218 adapter_image : PipelineImageInput = None ,
@@ -1224,11 +1224,11 @@ def __call__(
12241224 denoising_start : Optional [float ] = None ,
12251225 denoising_end : Optional [float ] = None ,
12261226 guidance_scale : float = 5.0 ,
1227- negative_prompt : Optional [Union [str , list [str ]]] = None ,
1228- negative_prompt_2 : Optional [Union [str , list [str ]]] = None ,
1227+ negative_prompt : Optional [Union [str , List [str ]]] = None ,
1228+ negative_prompt_2 : Optional [Union [str , List [str ]]] = None ,
12291229 num_images_per_prompt : Optional [int ] = 1 ,
12301230 eta : float = 0.0 ,
1231- generator : Optional [Union [torch .Generator , list [torch .Generator ]]] = None ,
1231+ generator : Optional [Union [torch .Generator , List [torch .Generator ]]] = None ,
12321232 latents : Optional [Union [torch .FloatTensor ]] = None ,
12331233 prompt_embeds : Optional [torch .FloatTensor ] = None ,
12341234 negative_prompt_embeds : Optional [torch .FloatTensor ] = None ,
@@ -1238,12 +1238,12 @@ def __call__(
12381238 return_dict : bool = True ,
12391239 callback : Optional [Callable [[int , int , torch .FloatTensor ], None ]] = None ,
12401240 callback_steps : int = 1 ,
1241- cross_attention_kwargs : Optional [dict [str , Any ]] = None ,
1241+ cross_attention_kwargs : Optional [Dict [str , Any ]] = None ,
12421242 guidance_rescale : float = 0.0 ,
1243- original_size : Optional [tuple [int , int ]] = None ,
1244- crops_coords_top_left : Optional [tuple [int , int ]] = (0 , 0 ),
1245- target_size : Optional [tuple [int , int ]] = None ,
1246- adapter_conditioning_scale : Optional [Union [float , list [float ]]] = 1.0 ,
1243+ original_size : Optional [Tuple [int , int ]] = None ,
1244+ crops_coords_top_left : Optional [Tuple [int , int ]] = (0 , 0 ),
1245+ target_size : Optional [Tuple [int , int ]] = None ,
1246+ adapter_conditioning_scale : Optional [Union [float , List [float ]]] = 1.0 ,
12471247 cond_tau : float = 1.0 ,
12481248 aesthetic_score : float = 6.0 ,
12491249 negative_aesthetic_score : float = 2.5 ,
0 commit comments