Commit
·
59660ba
1
Parent(s):
b426c55
deleting an assert that should be false
Browse files
tests.py
CHANGED
|
@@ -21,7 +21,6 @@ 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
|
| 27 |
|
|
|
|
| 21 |
pipeline: GroupedSamplingPipeLine = create_pipeline("gpt2")
|
| 22 |
assert pipeline is not None
|
| 23 |
assert pipeline.model_name == "gpt2"
|
|
|
|
| 24 |
assert pipeline.wrapped_model.end_of_sentence_stop is False
|
| 25 |
del pipeline
|
| 26 |
|