ermu2001 commited on
Commit
ba8b44a
1 Parent(s): 3660ce4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -1,18 +1,20 @@
1
  import sys
2
  from huggingface_hub import snapshot_download
3
  snapshot_download(
4
- 'ermu2001/pllava-13b',
5
- local_dir='MODELS/pllava-13b',
6
  repo_type='model',
7
  local_dir_use_symlinks=True,
8
  )
9
 
10
  sys.argv.extend([
11
- "--pretrained_model_name_or_path", "MODELS/pllava-13b",
12
  "--num_frames", "16",
13
  "--use_lora",
14
- "--weight_dir", "MODELS/pllava-13b",
15
  "--lora_alpha", "4",
16
- "--conv_mode", "plain",
 
17
  ])
 
18
  import tasks.eval.demo.pllava_demo
 
1
  import sys
2
  from huggingface_hub import snapshot_download
3
  snapshot_download(
4
+ 'ermu2001/pllava-34b',
5
+ local_dir='MODELS/pllava-34b',
6
  repo_type='model',
7
  local_dir_use_symlinks=True,
8
  )
9
 
10
  sys.argv.extend([
11
+ "--pretrained_model_name_or_path", "MODELS/pllava-34b",
12
  "--num_frames", "16",
13
  "--use_lora",
14
+ "--weight_dir", "MODELS/pllava-34b",
15
  "--lora_alpha", "4",
16
+ "--conv_mode", "eval_vcg_llavanext",
17
+ "--use_multi_gpus"
18
  ])
19
+
20
  import tasks.eval.demo.pllava_demo