Update llm_utils to check llama use

microsoft · Dec 20, 2024 · bda45b2 · bda45b2
1 parent d64a228
commit bda45b2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rdagent/oai/llm_utils.py b/rdagent/oai/llm_utils.py
@@ -44,7 +44,8 @@ def md5_hash(input_string: str) -> str:
     logger.warning("openai is not installed.")
 
 try:
-    from llama import Llama
+    if LLM_SETTINGS.use_llama2:
+        from llama import Llama
 except ImportError:
     logger.warning("llama is not installed.")