winglian commited on
Commit
a375959
1 Parent(s): 22b462e

hardcode model and paths for now...

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ from llama_lora.utils.data import init_data_dir
11
 
12
  def main(
13
  load_8bit: bool = False,
14
- base_model: str = "",
15
- data_dir: str = "",
16
  # Allows to listen on all interfaces by providing '0.0.0.0'.
17
  server_name: str = "127.0.0.1",
18
  share: bool = False,
 
11
 
12
  def main(
13
  load_8bit: bool = False,
14
+ base_model: str = "decapoda-research/llama-7b-hf",
15
+ data_dir: str = "./datasets",
16
  # Allows to listen on all interfaces by providing '0.0.0.0'.
17
  server_name: str = "127.0.0.1",
18
  share: bool = False,