yonikremer
commited on
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 |
|