Преглед на файлове

Updated model names for OctoAI

varunfb преди 11 месеца
родител
ревизия
eeb45e5f2c
променени са 1 файла, в които са добавени 3 реда и са изтрити 2 реда
  1. 3 2
      src/llama_recipes/inference/llm.py

+ 3 - 2
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",        
         ]