Former-commit-id: 7899b44b19c3d0a70706d987bb7d2e0e3536014b
This commit is contained in:
hiyouga 2024-09-05 02:49:22 +08:00
parent 571a9b8669
commit 7b01c0676c
2 changed files with 7 additions and 1 deletions

View File

@ -29,7 +29,7 @@ jobs:
os:
- "ubuntu-latest"
- "windows-latest"
- "macos-latest"
- "macos-12"
runs-on: ${{ matrix.os }}
@ -38,6 +38,7 @@ jobs:
env:
HF_TOKEN: ${{ secrets.HF_TOKEN }}
CI_OS: ${{ matrix.os }}
steps:
- name: Checkout

View File

@ -14,6 +14,8 @@
import os
import pytest
from llamafactory.train.test_utils import compare_model, load_infer_model, load_reference_model, load_train_model
@ -47,6 +49,8 @@ INFER_ARGS = {
"infer_dtype": "float16",
}
CI_OS = os.environ.get("CI_OS", "")
def test_pissa_train():
model = load_train_model(**TRAIN_ARGS)
@ -54,6 +58,7 @@ def test_pissa_train():
compare_model(model, ref_model)
@pytest.mark.skipif(CI_OS.startswith("windows"), reason="Skip for windows.")
def test_pissa_inference():
model = load_infer_model(**INFER_ARGS)
ref_model = load_reference_model(TINY_LLAMA_PISSA, TINY_LLAMA_PISSA, use_pissa=True, is_trainable=False)