leofan commited on
Commit
edb6021
·
verified ·
1 Parent(s): ac7bb48

Update TRANSFORMERS_CACHE environment variable

Browse files
Files changed (1) hide show
  1. app/server.py +2 -0
app/server.py CHANGED
@@ -11,6 +11,8 @@ from langchain.embeddings.huggingface import HuggingFaceEmbeddings
11
 
12
  app = FastAPI()
13
 
 
 
14
  hf_llm = HuggingFaceHub(
15
  repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
16
  huggingfacehub_api_token=os.environ["HF_TOKEN"],
 
11
 
12
  app = FastAPI()
13
 
14
+ os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
15
+
16
  hf_llm = HuggingFaceHub(
17
  repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
18
  huggingfacehub_api_token=os.environ["HF_TOKEN"],