kshitijk
commited on
Commit
•
6afa145
1
Parent(s):
a5cdc7a
Change readme and app
Browse files
README.md
CHANGED
@@ -8,8 +8,8 @@ sdk_version: 1.31.1
|
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
preload_from_hub:
|
11 |
-
|
12 |
-
- NousResearch/Llama-2-7b-chat-hf
|
13 |
- BAAI/bge-small-en-v1.5
|
14 |
---
|
15 |
|
|
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
preload_from_hub:
|
11 |
+
- NousResearch/Llama-2-13b-chat-hf model-00001-of-00003.safetensors, model-00002-of-00003.safetensors, model-00003-of-00003.safetensors
|
12 |
+
# - NousResearch/Llama-2-7b-chat-hf
|
13 |
- BAAI/bge-small-en-v1.5
|
14 |
---
|
15 |
|
app.py
CHANGED
@@ -25,7 +25,7 @@ default_bnb_config = BitsAndBytesConfig(
|
|
25 |
logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
|
26 |
logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
|
27 |
set_global_tokenizer(
|
28 |
-
AutoTokenizer.from_pretrained("NousResearch/Llama-2-
|
29 |
)
|
30 |
|
31 |
|
@@ -46,8 +46,8 @@ def getVectorIndex(docs):
|
|
46 |
|
47 |
def getLLM():
|
48 |
|
49 |
-
|
50 |
-
model_path = "NousResearch/Llama-2-7b-chat-hf"
|
51 |
|
52 |
llm = HuggingFaceLLM(
|
53 |
context_window=3900,
|
|
|
25 |
logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
|
26 |
logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
|
27 |
set_global_tokenizer(
|
28 |
+
AutoTokenizer.from_pretrained("NousResearch/Llama-2-13b-chat-hf").encode
|
29 |
)
|
30 |
|
31 |
|
|
|
46 |
|
47 |
def getLLM():
|
48 |
|
49 |
+
model_path = "NousResearch/Llama-2-13b-chat-hf"
|
50 |
+
# model_path = "NousResearch/Llama-2-7b-chat-hf"
|
51 |
|
52 |
llm = HuggingFaceLLM(
|
53 |
context_window=3900,
|