zetavg commited on
Commit
194723b
1 Parent(s): 1583e8c
Files changed (1) hide show
  1. app_hf_ui_demo.py +2 -1
app_hf_ui_demo.py CHANGED
@@ -11,7 +11,8 @@ from llama_lora.utils.data import init_data_dir
11
 
12
  def main():
13
  data_dir = os.path.abspath("./data")
14
- Global.default_base_model_name = "decapoda-research/llama-7b-hf"
 
15
  Global.data_dir = data_dir
16
  Global.load_8bit = False
17
 
 
11
 
12
  def main():
13
  data_dir = os.path.abspath("./data")
14
+ Global.default_base_model_name = Global.base_model_name = "decapoda-research/llama-7b-hf"
15
+ Global.base_model_choices = ["decapoda-research/llama-7b-hf", "nomic-ai/gpt4all-j"]
16
  Global.data_dir = data_dir
17
  Global.load_8bit = False
18