mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-23 03:11:19 +08:00
[CI] FIx fixture 'siglip_attention_config' not found (#31053)
Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
This commit is contained in:
parent
0be149524c
commit
ff2168bca3
18
tests/models/multimodal/pooling/conftest.py
Normal file
18
tests/models/multimodal/pooling/conftest.py
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
# SPDX-License-Identifier: Apache-2.0
|
||||||
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
||||||
|
"""Pytest configuration for vLLM pooling tests."""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from vllm.platforms import current_platform
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def siglip_attention_config():
|
||||||
|
"""Return attention config for SigLIP tests on ROCm.
|
||||||
|
|
||||||
|
On ROCm, SigLIP tests require FLEX_ATTENTION backend.
|
||||||
|
"""
|
||||||
|
if current_platform.is_rocm():
|
||||||
|
return {"backend": "FLEX_ATTENTION"}
|
||||||
|
return None
|
||||||
Loading…
x
Reference in New Issue
Block a user