testing create_pipeline with gpt2 and not with AVAILABLE MODELS to enable testing with low-memory

tests.py

@@ -17,14 +17,10 @@ def test_on_form_submit():
         on_form_submit(model_name, output_length, empty_prompt)
 
 
-@pytest.mark.parametrize(
-    "model_name",
-    AVAILABLE_MODELS,
-)
-def test_create_pipeline(model_name: str):
-    pipeline: GroupedSamplingPipeLine = create_pipeline(model_name)
+def test_create_pipeline():
+    pipeline: GroupedSamplingPipeLine = create_pipeline("gpt2")
     assert pipeline is not None
-    assert pipeline.model_name == model_name
+    assert pipeline.model_name == "gpt2"
     assert pipeline.wrapped_model.group_size == 5
     assert pipeline.wrapped_model.end_of_sentence_stop is False
     del pipeline