modify the logic of the condition

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