Use smaller model for unit tests (#315)

deepspeedai · Nov 16, 2023 · 2d49bc5 · 2d49bc5
1 parent 3d5c2d9
commit 2d49bc5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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