diff --git a/llm.py b/llm.py new file mode 100644 index 00000000..1906ca7c --- /dev/null +++ b/llm.py @@ -0,0 +1,5 @@ +# example +from swarms.utils.llm import LLM +llm_instance = LLM(hf_repo_id="google/flan-t5-xl", hf_api_token="your_hf_api_token") +result = llm_instance.run("Who won the FIFA World Cup in 1998?") +print(result) \ No newline at end of file diff --git a/swarms/utils/llm.py b/swarms/utils/llm.py index a77e1564..6fa46e0f 100644 --- a/swarms/utils/llm.py +++ b/swarms/utils/llm.py @@ -2,7 +2,9 @@ from typing import Optional import os import logging -from langchain import PromptTemplate, HuggingFaceHub, ChatOpenAI, LLMChain +from langchain import PromptTemplate, LLMChain, HuggingFaceHub +from langchain.chat_models import ChatOpenAI + # Configure logging- logging.basicConfig(level=logging.INFO)