kshitijk commited on
Commit
6afa145
1 Parent(s): a5cdc7a

Change readme and app

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. app.py +3 -3
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
- # - NousResearch/Llama-2-13b-chat-hf
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-7b-chat-hf").encode
29
  )
30
 
31
 
@@ -46,8 +46,8 @@ def getVectorIndex(docs):
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,
 
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,