shoom013 commited on
Commit
cbcd2cc
·
verified ·
1 Parent(s): 2da2d21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -11,7 +11,7 @@ from llama_index.core import (
11
  Document,
12
  Settings,
13
  )
14
- from llama_index.llms.huggingface_api import (HuggingFaceLLM, )
15
  from llama_index.llms.huggingface_api import (HuggingFaceInferenceAPI, )
16
  from llama_index.core.base.llms.types import ChatMessage
17
  from huggingface_hub import login
@@ -77,7 +77,8 @@ Settings.llm = HuggingFaceInferenceAPI(model_name=LLM_NAME,
77
  "presence_penalty": presence_penalty, "frequency_penalty": frequency_penalty,
78
  "top_k": top_k, "do_sample": False},
79
  # tokenizer_kwargs={"max_length": 4096},
80
- tokenizer_name=LLM_NAME,
 
81
  )
82
  # "BAAI/bge-m3"
83
  Settings.embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2")
 
11
  Document,
12
  Settings,
13
  )
14
+ from llama_index.llms.huggingface import (HuggingFaceLLM, )
15
  from llama_index.llms.huggingface_api import (HuggingFaceInferenceAPI, )
16
  from llama_index.core.base.llms.types import ChatMessage
17
  from huggingface_hub import login
 
77
  "presence_penalty": presence_penalty, "frequency_penalty": frequency_penalty,
78
  "top_k": top_k, "do_sample": False},
79
  # tokenizer_kwargs={"max_length": 4096},
80
+ tokenizer_name=LLM_NAME,
81
+ provider=auto,
82
  )
83
  # "BAAI/bge-m3"
84
  Settings.embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2")