Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,11 +15,6 @@ from mplug_owl2.utils import (build_logger, server_error_msg,
|
|
15 |
from model_worker import ModelWorker
|
16 |
import hashlib
|
17 |
|
18 |
-
from modelscope.hub.snapshot_download import snapshot_download
|
19 |
-
model_dir = snapshot_download('damo/mPLUG-Owl2', cache_dir='./')
|
20 |
-
|
21 |
-
print(os.listdir('./'))
|
22 |
-
|
23 |
|
24 |
logger = build_logger("gradio_web_server_local", "gradio_web_server_local.log")
|
25 |
|
@@ -375,7 +370,7 @@ if __name__ == "__main__":
|
|
375 |
parser.add_argument("--concurrency-count", type=int, default=10)
|
376 |
parser.add_argument("--model-list-mode", type=str, default="once",
|
377 |
choices=["once", "reload"])
|
378 |
-
parser.add_argument("--model-path", type=str, default="
|
379 |
parser.add_argument("--device", type=str, default="cuda")
|
380 |
parser.add_argument("--load-8bit", action="store_true")
|
381 |
parser.add_argument("--load-4bit", action="store_true")
|
|
|
15 |
from model_worker import ModelWorker
|
16 |
import hashlib
|
17 |
|
|
|
|
|
|
|
|
|
|
|
18 |
|
19 |
logger = build_logger("gradio_web_server_local", "gradio_web_server_local.log")
|
20 |
|
|
|
370 |
parser.add_argument("--concurrency-count", type=int, default=10)
|
371 |
parser.add_argument("--model-list-mode", type=str, default="once",
|
372 |
choices=["once", "reload"])
|
373 |
+
parser.add_argument("--model-path", type=str, default="MAGAer13/mplug-owl2-llama2-7b")
|
374 |
parser.add_argument("--device", type=str, default="cuda")
|
375 |
parser.add_argument("--load-8bit", action="store_true")
|
376 |
parser.add_argument("--load-4bit", action="store_true")
|