Procházet zdrojové kódy

Updated model names for OctoAI (#524)

varunfb před 1 rokem
rodič
revize
67aedd8227
1 změnil soubory, kde provedl 3 přidání a 2 odebrání
  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",        
         ]