MAGAer13 commited on
Commit
84d7d01
β€’
1 Parent(s): 848ce1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
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="./mPLUG-Owl2")
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")