Przeglądaj źródła

Updated model names for OctoAI (#524)

varunfb 11 miesięcy temu
rodzic
commit
67aedd8227
1 zmienionych plików z 3 dodań i 2 usunięć
  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
     @override
     def valid_models(self) -> list[str]:
     def valid_models(self) -> list[str]:
         return [
         return [
-            "llamaguard-7b",
-            "llama-2-13b-chat",
+            "llamaguard-2-8b",
+            "meta-llama-3-8b-instruct",
+            "meta-llama-3-70b-instruct",        
         ]
         ]