derektan commited on
Commit
c2ef65c
·
1 Parent(s): 3015687

Changed defaullt model and precision

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,11 +23,11 @@ from utils.utils import (DEFAULT_IM_END_TOKEN, DEFAULT_IM_START_TOKEN,
23
 
24
  def parse_args(args):
25
  parser = argparse.ArgumentParser(description="LISA chat")
26
- parser.add_argument("--version", default="xinlai/LISA-13B-llama2-v1")
27
  parser.add_argument("--vis_save_path", default="./vis_output", type=str)
28
  parser.add_argument(
29
  "--precision",
30
- default="fp16",
31
  type=str,
32
  choices=["fp32", "bf16", "fp16"],
33
  help="precision for inference",
 
23
 
24
  def parse_args(args):
25
  parser = argparse.ArgumentParser(description="LISA chat")
26
+ parser.add_argument("--version", default="derektan95/LISA-AVS")
27
  parser.add_argument("--vis_save_path", default="./vis_output", type=str)
28
  parser.add_argument(
29
  "--precision",
30
+ default="bf16",
31
  type=str,
32
  choices=["fp32", "bf16", "fp16"],
33
  help="precision for inference",