baLLseM / model /tools /__init__.py
hqms's picture
add HF Key
799a14a
raw
history blame contribute delete
261 Bytes
from langchain.chains import LLMChain
from langchain.agents import Tool
from langchain import HuggingFaceHub
import os
hf_auth = os.environ.get("HF_KEY")
llm = HuggingFaceHub(repo_id = "microsoft/Phi-3-mini-128k-instruct", huggingfacehub_api_token = hf_auth)