core.models.model_registry

core.models.model_registry#

Copyright (c) Meta, Inc. and its affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Attributes#

MODEL_REGISTRY

available_pretrained_models

Functions#

model_name_to_local_file(→ str)

Download a pretrained checkpoint if it does not exist already

Module Contents#

core.models.model_registry.MODEL_REGISTRY#
core.models.model_registry.available_pretrained_models#
core.models.model_registry.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

Contents