Files
OpenLLM/openllm-python/tests/conftest.py
Aaron Pham 97d76eec85 tests: add additional basic testing (#982)
* chore: update rebase tests

Signed-off-by: paperspace <29749331+aarnphm@users.noreply.github.com>

* chore: update partial clients before removing

Signed-off-by: paperspace <29749331+aarnphm@users.noreply.github.com>

* fix: update clients parsing logics to work with 0.5

Signed-off-by: paperspace <29749331+aarnphm@users.noreply.github.com>

* chore: ignore ci runs as to run locally

Signed-off-by: paperspace <29749331+aarnphm@users.noreply.github.com>

* chore: update async client tests

Signed-off-by: paperspace <29749331+aarnphm@users.noreply.github.com>

* chore: update pre-commit

Signed-off-by: paperspace <29749331+aarnphm@users.noreply.github.com>

---------

Signed-off-by: paperspace <29749331+aarnphm@users.noreply.github.com>
2024-05-23 10:02:23 -04:00

17 lines
363 B
Python

from __future__ import annotations
import pytest, typing as t
@pytest.fixture(
scope='function',
name='model_id',
params={
'meta-llama/Meta-Llama-3-8B-Instruct',
'casperhansen/llama-3-70b-instruct-awq',
'TheBloke/Nous-Hermes-2-Mixtral-8x7B-DPO-AWQ',
},
)
def fixture_model_id(request) -> t.Generator[str, None, None]:
yield request.param