Commit 90e10dee authored by zhuwenwen's avatar zhuwenwen
Browse files

[tests] update models

parent d7d4623e
This diff is collapsed.
......@@ -13,9 +13,11 @@ Run `pytest tests/models/test_bitblas.py`.
"""
from dataclasses import dataclass
import os
import pytest
from .utils import check_logprobs_close
from ..utils import models_path_prefix
@dataclass
......@@ -24,7 +26,7 @@ class ModelPair:
model_pairs = [
ModelPair(model_gptq="hxbgsyxh/opt-125m-4bit-128g"),
ModelPair(model_gptq=os.path.join(models_path_prefix, "hxbgsyxh/opt-125m-4bit-128g")),
]
......@@ -58,4 +60,4 @@ def test_models(
outputs_1_lst=bitblas_outputs,
name_0="gptq",
name_1="gptq_bitblas",
)
)
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment