def __init__()

in optimum/onnxruntime/modeling_diffusion.py [0:0]

• 21 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(
        self,
        *,
        # pipeline models
        unet_session: Optional["InferenceSession"] = None,
        transformer_session: Optional["InferenceSession"] = None,
        vae_decoder_session: Optional["InferenceSession"] = None,
        vae_encoder_session: Optional["InferenceSession"] = None,
        text_encoder_session: Optional["InferenceSession"] = None,
        text_encoder_2_session: Optional["InferenceSession"] = None,
        text_encoder_3_session: Optional["InferenceSession"] = None,
        # pipeline submodels
        scheduler: Optional["SchedulerMixin"] = None,
        tokenizer: Optional["CLIPTokenizer"] = None,
        tokenizer_2: Optional["CLIPTokenizer"] = None,
        tokenizer_3: Optional["CLIPTokenizer"] = None,
        feature_extractor: Optional["CLIPFeatureExtractor"] = None,
        # stable diffusion xl specific arguments
        force_zeros_for_empty_prompt: bool = True,
        requires_aesthetics_score: bool = False,
        add_watermarker: Optional[bool] = None,
        # onnxruntime specific arguments
        use_io_binding: Optional[bool] = None,
        model_save_dir: Optional[Union[str, Path, TemporaryDirectory]] = None,
        **kwargs,