fastvideo.v1.models.loader.utils

Utilities for selecting and loading models.

Module Contents

Functions

get_param_names_mapping	Creates a mapping function that transforms parameter names using regex patterns.
set_default_torch_dtype	Sets the default torch dtype to the given dtype.

Data

logger

API

fastvideo.v1.models.loader.utils.get_param_names_mapping(mapping_dict: Dict[str, str]) → Callable[[str], tuple[str, Any, Any]]

Creates a mapping function that transforms parameter names using regex patterns.

Parameters:

• mapping_dict (Dict[str, str]) – Dictionary mapping regex patterns to replacement patterns

• param_name (str) – The parameter name to be transformed

Returns:

A function that maps parameter names from source to target format

Return type:

Callable[[str], str]

fastvideo.v1.models.loader.utils.logger

‘init_logger(…)’

fastvideo.v1.models.loader.utils.set_default_torch_dtype(dtype: torch.dtype)

Sets the default torch dtype to the given dtype.