From df18590934bb3acf6736105bd1ff36793c65d562 Mon Sep 17 00:00:00 2001 From: Kye Date: Tue, 18 Jul 2023 11:51:40 -0400 Subject: [PATCH] clean up Former-commit-id: b0a609641c82dcbdff91e36267de47010c102980 --- swarms/utils/llm.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/swarms/utils/llm.py b/swarms/utils/llm.py index f7da10b4..31343b70 100644 --- a/swarms/utils/llm.py +++ b/swarms/utils/llm.py @@ -8,6 +8,7 @@ from langchain import PromptTemplate, HuggingFaceHub, ChatOpenAI, LLMChain logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) + class LLM: def __init__(self, openai_api_key: Optional[str] = None, @@ -30,9 +31,9 @@ class LLM: if self.hf_api_token: os.environ["HUGGINGFACEHUB_API_TOKEN"] = self.hf_api_token - #initialize the LLM project + # Initialize the LLM object self.initialize_llm() - + def initialize_llm(self): model_kwargs = {"temperature": self.temperature, "max_length": self.max_length} try: