Former-commit-id: 70078d8845ab5af847bffbe5192fad563ffabb3e
pull/160/head
Kye 2 years ago
parent 71fe56159c
commit 40a200c924

@ -52,13 +52,13 @@ class LLM:
logger.error("Failed to generate response: %s", e) logger.error("Failed to generate response: %s", e)
raise raise
# example # # example
from swarms.utils.llm import LLM # from swarms.utils.llm import LLM
llm_instance = LLM(openai_api_key="your_openai_key") # llm_instance = LLM(openai_api_key="your_openai_key")
result = llm_instance.run("Who won the FIFA World Cup in 1998?") # result = llm_instance.run("Who won the FIFA World Cup in 1998?")
print(result) # print(result)
# using HuggingFaceHub # # using HuggingFaceHub
llm_instance = LLM(hf_repo_id="google/flan-t5-xl", hf_api_token="your_hf_api_token") # 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?") # result = llm_instance.run("Who won the FIFA World Cup in 1998?")
print(result) # print(result)

Loading…
Cancel
Save