Skip to content

Commit cdd113b

Browse files
committed
feat: support vlm do dynamic compile for vit
- vit can be dynamic compiled by option "--dynamic_vit" Change-Id: Iecde709825bfb40afa17569f0ca4906c620c8332
1 parent ac4f724 commit cdd113b

2 files changed

Lines changed: 6 additions & 1 deletion

File tree

python/llm/LlmConverter.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ def __init__(self, args, config):
4141
self.chip = args.chip
4242
self.embedding_disk = args.embedding_disk
4343
self.dynamic = args.dynamic
44+
self.dynamic_vit = args.dynamic_vit
4445
self.use_block_with_kv = args.use_block_with_kv
4546
self.debug = args.debug
4647
self.position_shape = [1, self.max_input_length]
@@ -1392,6 +1393,8 @@ def compile_vit(self):
13921393
deploy_args.append('--high_precision')
13931394
if self.debug:
13941395
deploy_args.append('--debug')
1396+
if self.dynamic_vit:
1397+
deploy_args.append('--dynamic')
13951398
self.add_task(deploy_args, f"{name}.log")
13961399

13971400
def combine(self):

python/tools/llm_convert.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,9 @@ def parse_max_pixels(value):
7070
parser.add_argument('--max_pixels', type=parse_max_pixels, default=0,
7171
help="max pixels for vit, for example: 240,420 or 100800")
7272
parser.add_argument('--dynamic', action='store_true',
73-
help='enable dynamic compiling for prefill, not recommended')
73+
help='enable dynamic compiling for llm prefill')
74+
parser.add_argument("--dynamic_vit", action='store_true',
75+
help='enable dynamic compiling for vit')
7476
parser.add_argument('--debug', action='store_true',
7577
help='enable debug mode, temp files will not be deleted')
7678
parser.add_argument("--again", action='store_true',

0 commit comments

Comments
 (0)