Unverified Commit 996aa70f authored by Isotr0py's avatar Isotr0py Committed by GitHub
Browse files

[Bugfix] Fix broken phi3-v mm_processor_kwargs tests (#11263)


Signed-off-by: default avatarIsotr0py <2037008807@qq.com>
parent 60508ffd
...@@ -58,16 +58,14 @@ def test_max_tokens_override(get_max_phi3v_image_tokens, model: str, ...@@ -58,16 +58,14 @@ def test_max_tokens_override(get_max_phi3v_image_tokens, model: str,
@pytest.mark.parametrize("model", models) @pytest.mark.parametrize("model", models)
@pytest.mark.parametrize( @pytest.mark.parametrize(
"num_crops,expected_toks_per_img,num_imgs", "num_crops,expected_toks_per_img",
[ [
(4, 757, 1), (4, 757),
(4, 757, 2), (16, 1921),
(16, 1921, 1),
(16, 1921, 2),
# the default num_crops of phi-3.5-vision is 4 # the default num_crops of phi-3.5-vision is 4
(None, 757, 2), (None, 757),
(None, 757, 2),
]) ])
@pytest.mark.parametrize("num_imgs", [1, 2])
def test_processor_override(processor_for_phi3v, image_assets: _ImageAssets, def test_processor_override(processor_for_phi3v, image_assets: _ImageAssets,
model: str, num_crops: Optional[int], model: str, num_crops: Optional[int],
expected_toks_per_img: int, num_imgs: int): expected_toks_per_img: int, num_imgs: int):
......
...@@ -302,11 +302,18 @@ class Phi3HDImageEmbedding(Phi3ImageEmbeddingBase): ...@@ -302,11 +302,18 @@ class Phi3HDImageEmbedding(Phi3ImageEmbeddingBase):
return image_features_hd_newline return image_features_hd_newline
def get_max_phi3v_image_tokens(ctx: InputContext) -> int: def get_max_phi3v_image_tokens(
processor = ctx.get_hf_processor() ctx: InputContext,
image_processor = processor.image_processor # type: ignore *,
num_crops: Optional[int] = None,
return image_processor.calc_num_image_tokens_from_image_size( ) -> int:
mm_processor_kwargs = {}
if num_crops:
mm_processor_kwargs["num_crops"] = num_crops
processor = ctx.get_hf_processor(**mm_processor_kwargs)
return processor.calc_num_image_tokens_from_image_size(
width=MAX_IMAGE_FEATURE_SIZE_WIDTH, width=MAX_IMAGE_FEATURE_SIZE_WIDTH,
height=MAX_IMAGE_FEATURE_SIZE_HEIGHT, height=MAX_IMAGE_FEATURE_SIZE_HEIGHT,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment