Skip to content

Commit f27347b

Browse files
committed
fix
1 parent 32c7ebd commit f27347b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

lightllm/utils/shm_size_check.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def _check_shm_size(args):
3030
shm_size = _get_system_shm_size_gb()
3131
required_size = _get_recommended_shm_size_gb(args)
3232
if shm_size < required_size:
33-
logger.warning(f"{RED}Available shm size is less than {shm_size:.2f}G{ENDC}")
33+
logger.warning(f"{RED}Available shm size {shm_size:.2f}G is less than required_size {required_size:.2f}G{ENDC}")
3434
return shm_size, required_size, False
3535
else: # shm_size >= required_size
3636
return shm_size, required_size, True
@@ -105,7 +105,9 @@ def _get_recommended_shm_size_gb(args, max_image_resolution=(3940, 2160), dtype_
105105

106106
# 假设加载最大分辨率图片时,通过 tokenizer 得到最多的 image_tokens
107107
if not hasattr(tokenizer, "get_image_token_length"):
108-
raise AttributeError("Tokenizer must have a 'get_image_token_length' method for multimodal models.")
108+
logger.error("Tokenizer must have a 'get_image_token_length' method for multimodal models.")
109+
return float("inf")
110+
109111
fake_image_item = ImageItem(
110112
type="image_size",
111113
data=max_image_resolution,

0 commit comments

Comments
 (0)