def __init__()

in optimum/furiosa/modeling_base.py [0:0]

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

    def __init__(
        self,
        model: Union[bytes, str, Path],
        config: PretrainedConfig = None,
        device: str = None,
        furiosa_config: Optional[Dict[str, str]] = None,
        model_save_dir: Optional[Union[str, Path, TemporaryDirectory]] = None,
        input_shape_dict: Optional[Dict[str, Tuple[int]]] = None,
        output_shape_dict: Optional[Dict[str, Tuple[int]]] = None,
        **kwargs,