alessandro trinca tornidor commited on
Commit
a5b4be9
·
1 Parent(s): c41e6ce

[feat] load LISA model in 4 bit to avoid Memory limit exceeded errors

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -52,7 +52,7 @@ def health() -> str:
52
  @session_logger.set_uuid_logging
53
  def parse_args(args_to_parse):
54
  parser = argparse.ArgumentParser(description="LISA chat")
55
- parser.add_argument("--version", default="xinlai/LISA-13B-llama2-v1")
56
  parser.add_argument("--vis_save_path", default="./vis_output", type=str)
57
  parser.add_argument(
58
  "--precision",
@@ -69,7 +69,7 @@ def parse_args(args_to_parse):
69
  )
70
  parser.add_argument("--local-rank", default=0, type=int, help="node rank")
71
  parser.add_argument("--load_in_8bit", action="store_true", default=False)
72
- parser.add_argument("--load_in_4bit", action="store_true", default=False)
73
  parser.add_argument("--use_mm_start_end", action="store_true", default=True)
74
  parser.add_argument(
75
  "--conv_type",
 
52
  @session_logger.set_uuid_logging
53
  def parse_args(args_to_parse):
54
  parser = argparse.ArgumentParser(description="LISA chat")
55
+ parser.add_argument("--version", default="xinlai/LISA-13B-llama2-v1-explanatory")
56
  parser.add_argument("--vis_save_path", default="./vis_output", type=str)
57
  parser.add_argument(
58
  "--precision",
 
69
  )
70
  parser.add_argument("--local-rank", default=0, type=int, help="node rank")
71
  parser.add_argument("--load_in_8bit", action="store_true", default=False)
72
+ parser.add_argument("--load_in_4bit", action="store_true", default=True)
73
  parser.add_argument("--use_mm_start_end", action="store_true", default=True)
74
  parser.add_argument(
75
  "--conv_type",