def _from_pretrained()

in src/optimum/nvidia/hub.py [0:0]

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

    def _from_pretrained(
        cls: Type[T],
        *,
        model_id: str,
        config: Dict,
        revision: Optional[str],
        cache_dir: Optional[Union[str, Path]],
        force_download: bool,
        proxies: Optional[Dict],
        resume_download: bool,
        local_files_only: bool,
        token: Optional[Union[str, bool]],
        use_cuda_graph: bool = False,
        device_map: Optional[str] = None,
        export_config: Optional[ExportConfig] = None,
        quantization_config: Optional[ModelOptRecipe] = None,
        force_export: bool = False,
        export_only: bool = False,
        save_intermediate_checkpoints: bool = False,