Skip to content

Commit eeb45e5

Browse files
committed
Updated model names for OctoAI
1 parent 519c5a6 commit eeb45e5

File tree

1 file changed

+3
-2
lines changed
  • src/llama_recipes/inference

1 file changed

+3
-2
lines changed

src/llama_recipes/inference/llm.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,7 @@ def query(self, prompt: str) -> str:
187187
@override
188188
def valid_models(self) -> list[str]:
189189
return [
190-
"llamaguard-7b",
191-
"llama-2-13b-chat",
190+
"llamaguard-2-8b",
191+
"meta-llama-3-8b-instruct",
192+
"meta-llama-3-70b-instruct",
192193
]

0 commit comments

Comments
 (0)