Skip to content

Commit 6ea57ef

Browse files
committed
[add]add disable_prompt_cache
1 parent 13b0c83 commit 6ea57ef

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

lightllm/utils/shm_size_check.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,7 @@ def _get_recommended_shm_size_gb(args, max_image_resolution=(3940, 2160), dtype_
117117
)
118118
fake_image_item.image_w = fake_image_item._data[0]
119119
fake_image_item.image_h = fake_image_item._data[1]
120+
# for internvl model shm check
120121
fake_image_item.extra_params["image_patch_max_num"] = 12
121122
max_image_tokens = tokenizer.get_image_token_length(fake_image_item)
122123

0 commit comments

Comments
 (0)