diff --git a/src/llama_recipes/inference/llm.py b/src/llama_recipes/inference/llm.py index 56a1e2dfe02bb5624849be51bc4eb8d9d7478b68..22237badd955752b6f3972fbdbc2af1b59b6bfb9 100644 --- a/src/llama_recipes/inference/llm.py +++ b/src/llama_recipes/inference/llm.py @@ -187,6 +187,7 @@ class OctoAI(LLM): @override def valid_models(self) -> list[str]: return [ - "llamaguard-7b", - "llama-2-13b-chat", + "llamaguard-2-8b", + "meta-llama-3-8b-instruct", + "meta-llama-3-70b-instruct", ]