llama-stack-mirror/llama_toolchain/common/model_utils.py

8 lines
212 B
Python

import os
from llama_models.datatypes import Model
from .config_dirs import DEFAULT_CHECKPOINT_DIR
def model_local_dir(model: Model) -> str:
return os.path.join(DEFAULT_CHECKPOINT_DIR, model.descriptor())