Commit
•
b426c55
1
Parent(s):
2f94880
testing create_pipeline with gpt2 and not with AVAILABLE MODELS to enable testing with low-memory
Browse files
tests.py
CHANGED
@@ -17,14 +17,10 @@ def test_on_form_submit():
|
|
17 |
on_form_submit(model_name, output_length, empty_prompt)
|
18 |
|
19 |
|
20 |
-
|
21 |
-
"
|
22 |
-
AVAILABLE_MODELS,
|
23 |
-
)
|
24 |
-
def test_create_pipeline(model_name: str):
|
25 |
-
pipeline: GroupedSamplingPipeLine = create_pipeline(model_name)
|
26 |
assert pipeline is not None
|
27 |
-
assert pipeline.model_name ==
|
28 |
assert pipeline.wrapped_model.group_size == 5
|
29 |
assert pipeline.wrapped_model.end_of_sentence_stop is False
|
30 |
del pipeline
|
|
|
17 |
on_form_submit(model_name, output_length, empty_prompt)
|
18 |
|
19 |
|
20 |
+
def test_create_pipeline():
|
21 |
+
pipeline: GroupedSamplingPipeLine = create_pipeline("gpt2")
|
|
|
|
|
|
|
|
|
22 |
assert pipeline is not None
|
23 |
+
assert pipeline.model_name == "gpt2"
|
24 |
assert pipeline.wrapped_model.group_size == 5
|
25 |
assert pipeline.wrapped_model.end_of_sentence_stop is False
|
26 |
del pipeline
|