iamrobotbear commited on
Commit
4584ae8
1 Parent(s): fa5dc36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,8 @@ logging.basicConfig(
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
 
 
14
  format="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s",
15
  )
16
 
17
+ #base_model = "project-baize/baize-v2-7b"
18
+ base_model = "h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v2"
19
  adapter_model = None
20
  tokenizer,model,device = load_tokenizer_and_model(base_model,adapter_model)
21