random2222 commited on
Commit
3b4718c
·
verified ·
1 Parent(s): a1f651b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -11,6 +11,10 @@ from langchain_community.llms import HuggingFacePipeline
11
  from transformers import pipeline, AutoTokenizer, BitsAndBytesConfig
12
  from huggingface_hub import login
13
 
 
 
 
 
14
  # HF Token handling
15
  if os.environ.get("HF_TOKEN"):
16
  login(token=os.environ["HF_TOKEN"])
 
11
  from transformers import pipeline, AutoTokenizer, BitsAndBytesConfig
12
  from huggingface_hub import login
13
 
14
+
15
+
16
+ torch.set_num_threads(2)
17
+
18
  # HF Token handling
19
  if os.environ.get("HF_TOKEN"):
20
  login(token=os.environ["HF_TOKEN"])