Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ from llama_index.core import (
|
|
| 11 |
Document,
|
| 12 |
Settings,
|
| 13 |
)
|
| 14 |
-
from llama_index.llms.
|
| 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")
|