`neural_compressor.ux.components.model_zoo.downloader`¶

Download model from Examples.

Module Contents¶

Classes¶

UX model downloader class.

class neural_compressor.ux.components.model_zoo.downloader.Downloader(data: Dict[str, Any])¶

UX model downloader class.

download_config() → str¶: Find yaml config resource and initialize downloading.

download_yaml_config(model_info: Dict[str, Any]) → str¶: Download config from GitHub for specified model.

download_model() → str¶: Find model resource and initialize downloading.

download(model_info: Dict[str, Any]) → str¶: Download specified model.

download_file(url: str, download_path: str, headers: dict | None = {}, report_progress: bool | None = True) → None¶: Download specified file.

unpack_archive(archive_path: str, filename: str) → str¶: Unpack archive and return path to unpacked model.

get_yaml_url(model_src_dir: str, yaml_relative_location: str) → Tuple[str, dict]¶: Get url for yaml config download.