s: int = 7
generator: typing.Union[torch._C.Generator, typing.List[torch._C.Generator], NoneType] = None
prior_latents: typing.Optional[torch.FloatTensor] = None
decoder_latents: typing.Optional[torch.FloatTensor] = None
super_res_latents: typing.Optional[torch.FloatTensor] = None
text_model_output: typing.Union[transformers.models.clip.modeling_clip.CLIPTextModelOutput, typing.Tuple, NoneType] = None
text_attention_mask: typing.Optional[torch.