conftest.py 331 Bytes
Newer Older
1
2
3
4
5
6
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project

import pytest
from transformers import AutoTokenizer

7
from vllm.tokenizers import TokenizerLike
8
9
10


@pytest.fixture(scope="function")
11
def default_tokenizer() -> TokenizerLike:
12
    return AutoTokenizer.from_pretrained("gpt2")