From eeb45e5f2c67c4e306d339d3612d48d07039d7e4 Mon Sep 17 00:00:00 2001
From: varunfb <vontimitta@fb.com>
Date: Fri, 17 May 2024 10:24:44 -0700
Subject: [PATCH] Updated model names for OctoAI
---
src/llama_recipes/inference/llm.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/llama_recipes/inference/llm.py b/src/llama_recipes/inference/llm.py
index 56a1e2df..22237bad 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",
]
--
GitLab