project-baize commited on
Commit
d6a22b6
1 Parent(s): 1abe38b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ logging.basicConfig(
14
  format="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s",
15
  )
16
 
17
- base_model = "decapoda-research/llama-7b-hf"
18
- adapter_model = "project-baize/baize-lora-7B"
19
  tokenizer,model,device = load_tokenizer_and_model(base_model,adapter_model)
20
 
21
  total_count = 0
 
14
  format="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s",
15
  )
16
 
17
+ base_model = "project-baize/baize-v2-7b"
18
+ adapter_model = None
19
  tokenizer,model,device = load_tokenizer_and_model(base_model,adapter_model)
20
 
21
  total_count = 0