sh-zam commited on
Commit
b56f08b
·
verified ·
1 Parent(s): bf1090f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -34,6 +34,7 @@ def start_worker(model_path: str, bits=16):
34
  assert bits in [4, 8, 16], "It can be only loaded with 16-bit, 8-bit, and 4-bit."
35
  if bits != 16:
36
  model_name += f"-{bits}bit"
 
37
  worker_command = [
38
  sys.executable,
39
  "-m",
@@ -46,6 +47,8 @@ def start_worker(model_path: str, bits=16):
46
  model_path,
47
  "--model-name",
48
  model_name,
 
 
49
  "--use-flash-attn",
50
  ]
51
  if bits != 16:
 
34
  assert bits in [4, 8, 16], "It can be only loaded with 16-bit, 8-bit, and 4-bit."
35
  if bits != 16:
36
  model_name += f"-{bits}bit"
37
+ model_name += "-lora"
38
  worker_command = [
39
  sys.executable,
40
  "-m",
 
47
  model_path,
48
  "--model-name",
49
  model_name,
50
+ "--model-base",
51
+ "llava-hf/llava-1.5-7b-hf"
52
  "--use-flash-attn",
53
  ]
54
  if bits != 16: