Kaynağa Gözat

Updated model names for OctoAI

varunfb 11 ay önce
ebeveyn
işleme
eeb45e5f2c
1 değiştirilmiş dosya ile 3 ekleme ve 2 silme
  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",        
         ]