def __init__()

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

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

    def __init__(
        self,
        config: Dict[str, Any],
        configs: Dict[str, "PretrainedConfig"],
        neuron_configs: Dict[str, "NeuronDefaultConfig"],
        data_parallel_mode: Literal["none", "unet", "transformer", "all"],
        scheduler: Optional[SchedulerMixin],
        vae_decoder: Union[torch.jit._script.ScriptModule, "NeuronModelVaeDecoder"],
        text_encoder: Optional[Union[torch.jit._script.ScriptModule, "NeuronModelTextEncoder"]] = None,
        text_encoder_2: Optional[Union[torch.jit._script.ScriptModule, "NeuronModelTextEncoder"]] = None,
        unet: Optional[Union[torch.jit._script.ScriptModule, "NeuronModelUnet"]] = None,
        transformer: Optional[Union[torch.jit._script.ScriptModule, "NeuronModelTransformer"]] = None,
        vae_encoder: Optional[Union[torch.jit._script.ScriptModule, "NeuronModelVaeEncoder"]] = None,
        image_encoder: Optional[torch.jit._script.ScriptModule] = None,
        safety_checker: Optional[torch.jit._script.ScriptModule] = None,
        tokenizer: Optional[Union[CLIPTokenizer, T5Tokenizer]] = None,
        tokenizer_2: Optional[CLIPTokenizer] = None,
        feature_extractor: Optional[CLIPFeatureExtractor] = None,
        controlnet: Optional[
            Union[
                torch.jit._script.ScriptModule,
                List[torch.jit._script.ScriptModule],
                "NeuronControlNetModel",
                "NeuronMultiControlNetModel",
            ]
        ] = None,
        # stable diffusion xl specific arguments
        requires_aesthetics_score: bool = False,
        force_zeros_for_empty_prompt: bool = True,
        add_watermarker: Optional[bool] = None,
        model_save_dir: Optional[Union[str, Path, TemporaryDirectory]] = None,
        model_and_config_save_paths: Optional[Dict[str, Tuple[str, Path]]] = None,