core.models#
Subpackages#
Submodules#
Attributes#
Functions#
|
Download a pretrained checkpoint if it does not exist already |
Package Contents#
- core.models.available_pretrained_models#
- core.models.model_name_to_local_file(model_name: str, local_cache: str | pathlib.Path) str #
Download a pretrained checkpoint if it does not exist already
- Parameters:
model_name (str) – the model name. See available_pretrained_checkpoints.
local_cache (str or Path) – path to local cache directory
- Returns:
local path to checkpoint file
- Return type:
str