Former-commit-id: 280c0f3f2cea4dfced797cc0e15f72b8b3a93542
This commit is contained in:
hiyouga
2024-09-05 03:02:59 +08:00
parent 7b01c0676c
commit 26d914b8fc
3 changed files with 4 additions and 3 deletions

View File

@@ -52,6 +52,7 @@ INFER_ARGS = {
CI_OS = os.environ.get("CI_OS", "")
@pytest.mark.skipif(CI_OS.startswith("windows"), reason="Skip for windows.")
def test_pissa_train():
model = load_train_model(**TRAIN_ARGS)
ref_model = load_reference_model(TINY_LLAMA_PISSA, TINY_LLAMA_PISSA, use_pissa=True, is_trainable=True)