llama-stack/llama_toolchain/common/model_utils.py
Dalton Flanagan b311dcd143 formatting
2024-08-14 17:03:43 -04:00

9 lines
213 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())