reduce default testing model from OPT-1.3B to OPT-125m

deepspeedai · Nov 16, 2023 · c43ab8e · c43ab8e
1 parent 3d5c2d9
commit c43ab8e
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