mirror of
https://github.com/deepseek-ai/DeepInfer.git
synced 2026-10-28 23:07:09 +08:00
17 lines
430 B
Python
17 lines
430 B
Python
"""Long-context HF regression, separate from the short CI workload."""
|
|
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from examples._config import DEFAULT_MODEL_NAME
|
|
from tests.models.test_offline_inference import _run_offline_inference
|
|
|
|
pytestmark = [
|
|
pytest.mark.long_context,
|
|
]
|
|
|
|
|
|
def test_offline_inference_long_decode_ppl() -> None:
|
|
_run_offline_inference(DEFAULT_MODEL_NAME, int(os.environ.get("DEVICE_OFFSET", "0")), long_context=True)
|