Spaces:
Build error
Build error
Update app.py
Browse files
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:
|