feat: Add testing for local models

pull/18/head
Zack 2 years ago
parent 404f230e35
commit c48f8574a9

@ -0,0 +1,6 @@
from swarms.utils.llm import LLM
# using HuggingFaceHub
llm_instance = LLM(hf_repo_id="Writer/camel-5b-hf", hf_api_token="your_hf_api_token")
result = llm_instance.run("Who won the FIFA World Cup in 1998?")
print(llm_instance)

@ -10,7 +10,7 @@ nest_asyncio
bs4 bs4
playwright playwright
duckduckgo-search duckduckgo_search
faiss-cpu faiss-cpu
wget==3.2 wget==3.2

Loading…
Cancel
Save