update llama version

huggingface · Aug 6, 2024 · 1dd3d7d · 1dd3d7d
1 parent 7a4973d
commit 1dd3d7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/agents/llm_engine.py b/src/transformers/agents/llm_engine.py
@@ -66,7 +66,7 @@ def get_clean_message_list(message_list: List[Dict[str, str]], role_conversions:
 
 
 class HfEngine:
-    def __init__(self, model: str = "meta-llama/Meta-Llama-3-8B-Instruct"):
+    def __init__(self, model: str = "meta-llama/Meta-Llama-3.1-8B-Instruct"):
         self.model = model
         self.client = InferenceClient(self.model, timeout=120)