|
44 | 44 |
|
45 | 45 |
|
46 | 46 | def test_empty_file(): |
| 47 | + from vllm.platforms import current_platform |
| 48 | + |
| 49 | + if current_platform.is_rocm(): |
| 50 | + pytest.skip( |
| 51 | + "intfloat/multilingual-e5-small is encoder-only, not supported on ROCm" |
| 52 | + ) |
| 53 | + |
47 | 54 | with ( |
48 | 55 | tempfile.NamedTemporaryFile("w") as input_file, |
49 | 56 | tempfile.NamedTemporaryFile("r") as output_file, |
@@ -128,6 +135,13 @@ def test_completions_invalid_input(): |
128 | 135 |
|
129 | 136 |
|
130 | 137 | def test_embeddings(): |
| 138 | + from vllm.platforms import current_platform |
| 139 | + |
| 140 | + if current_platform.is_rocm(): |
| 141 | + pytest.skip( |
| 142 | + "intfloat/multilingual-e5-small is encoder-only, not supported on ROCm" |
| 143 | + ) |
| 144 | + |
131 | 145 | with ( |
132 | 146 | tempfile.NamedTemporaryFile("w") as input_file, |
133 | 147 | tempfile.NamedTemporaryFile("r") as output_file, |
@@ -159,6 +173,11 @@ def test_embeddings(): |
159 | 173 |
|
160 | 174 | @pytest.mark.parametrize("input_batch", [INPUT_SCORE_BATCH, INPUT_RERANK_BATCH]) |
161 | 175 | def test_score(input_batch): |
| 176 | + from vllm.platforms import current_platform |
| 177 | + |
| 178 | + if current_platform.is_rocm(): |
| 179 | + pytest.skip("BAAI/bge-reranker-v2-m3 is encoder-only, not supported on ROCm") |
| 180 | + |
162 | 181 | with ( |
163 | 182 | tempfile.NamedTemporaryFile("w") as input_file, |
164 | 183 | tempfile.NamedTemporaryFile("r") as output_file, |
|
0 commit comments