Skip to content

Commit d9281ee

Browse files
authored
[FIX]avoid image_max_patch_num=0 (#902)
1 parent 384ff32 commit d9281ee

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lightllm/models/internvl/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def __init__(self, tokenizer, model_cfg, **kwargs):
4848
def init_imageitem_extral_params(
4949
self, img: ImageItem, multi_params: MultimodalParams, sampling_params: SamplingParams
5050
):
51-
if sampling_params.image_max_patch_num >= 0:
51+
if sampling_params.image_max_patch_num > 0:
5252
img.extra_params["image_patch_max_num"] = sampling_params.image_max_patch_num
5353
return
5454
elif os.getenv("MAX_PATCH_NUM"):

0 commit comments

Comments
 (0)