Added support for llama 3.3 model

This commit is contained in:
varunfb 2024-12-10 19:51:11 -08:00
parent 76eb558bde
commit 47c23bfc6e

View file

@ -27,7 +27,8 @@ def supported_inference_models() -> List[Model]:
m m
for m in all_registered_models() for m in all_registered_models()
if ( if (
m.model_family in {ModelFamily.llama3_1, ModelFamily.llama3_2} m.model_family
in {ModelFamily.llama3_1, ModelFamily.llama3_2, ModelFamily.llama3_3}
or is_supported_safety_model(m) or is_supported_safety_model(m)
) )
] ]