feat: introduce llama4 support

This commit is contained in:
Ashwin Bharambe 2025-04-05 09:39:05 -07:00
parent 23a99a4b22
commit 20cbcdf0d5
61 changed files with 205222 additions and 6439 deletions

View file

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