Spaces:
Paused
Paused
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
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=
|
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",
|