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`. ...@@ -13,9 +13,11 @@ Run `pytest tests/models/test_bitblas.py`.
""" """
from dataclasses import dataclass from dataclasses import dataclass
import os
import pytest import pytest
from .utils import check_logprobs_close from .utils import check_logprobs_close
from ..utils import models_path_prefix
@dataclass @dataclass
...@@ -24,7 +26,7 @@ class ModelPair: ...@@ -24,7 +26,7 @@ class ModelPair:
model_pairs = [ 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( ...@@ -58,4 +60,4 @@ def test_models(
outputs_1_lst=bitblas_outputs, outputs_1_lst=bitblas_outputs,
name_0="gptq", name_0="gptq",
name_1="gptq_bitblas", 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