diff options
| author | Ashwin Bharambe <ashwin@meta.com> | 2024-08-22 12:45:37 -0700 |
|---|---|---|
| committer | Ashwin Bharambe <ashwin@meta.com> | 2024-08-22 12:45:37 -0700 |
| commit | d7b79b0a44e3f384571478d6a2d2be6ce7602853 (patch) | |
| tree | e338504f9fe3db4053d8d9b8e0757bc3a72b14be /models | |
| parent | cb3240d00aaf0a564e1a97d9b21705374d26d323 (diff) | |
Another fix for llama-guard-2
Diffstat (limited to 'models')
| -rw-r--r-- | models/sku_list.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/models/sku_list.py b/models/sku_list.py index 661c856..801b965 100644 --- a/models/sku_list.py +++ b/models/sku_list.py @@ -618,7 +618,7 @@ def safety_models() -> List[Model]: ), Model( core_model_id=CoreModelId.llama_guard_2_8b, - is_default_variant=False, + is_default_variant=True, description_markdown="Llama Guard v2 8b system safety model", huggingface_repo="meta-llama/Meta-Llama-Guard-2-8B", hardware_requirements=HardwareRequirements( @@ -676,6 +676,8 @@ def llama_meta_net_info(model: Model) -> LlamaDownloadInfo: folder = "Meta-Llama-Guard-3-8B" elif model.core_model_id == CoreModelId.prompt_guard_86m: folder = "Prompt-Guard" + elif model.core_model_id == CoreModelId.llama_guard_2_8b: + folder = "llama-guard-2" else: folder = model.huggingface_repo.split("/")[-1] if "Llama-2" in folder: |