From 2d49bc5c52611c93a21e4d97fdeef3a78c34afce Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Thu, 16 Nov 2023 11:39:15 -0800 Subject: [PATCH] Use smaller model for unit tests (#315) --- tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index 808666cf..aeecb02d 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -43,7 +43,7 @@ def task_name(request): return request.param -@pytest.fixture(scope="function", params=["facebook/opt-1.3b"]) +@pytest.fixture(scope="function", params=["facebook/opt-125m"]) def model_name(request): return request.param