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
@ -52,7 +52,7 @@ ggl
gradio_tools gradio_tools
arxiv arxiv
google-api-python-client google-api-python-client
google-auth-httplib2 google-auth-httplib2
beautifulsoup4==4.11.2 beautifulsoup4==4.11.2
@ -104,4 +104,4 @@ timm
typeguard typeguard
pytesseract pytesseract
huggingface_hub huggingface_hub

Loading…
Cancel
Save