"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "eac5b8bfbdd1f1bd949b7dce2793b5afc5f64d02"
Unverified Commit e71f32c0 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[testing] fix ambiguous test (#6898)

Since `generate()` does:
```
        num_beams = num_beams if num_beams is not None else self.config.num_beams
```
This test fails if `model.config.num_beams > 1` (which is the case in the model I'm porting).

This fix makes the test setup unambiguous by passing an explicit `num_beams=1` to `generate()`.

Thanks.
parent 8f2723ca
...@@ -822,7 +822,7 @@ class ModelTesterMixin: ...@@ -822,7 +822,7 @@ class ModelTesterMixin:
with self.assertRaises(AssertionError): with self.assertRaises(AssertionError):
# generating multiple sequences when no beam search generation # generating multiple sequences when no beam search generation
# is not allowed as it would always generate the same sequences # is not allowed as it would always generate the same sequences
model.generate(input_ids, do_sample=False, num_return_sequences=2) model.generate(input_ids, do_sample=False, num_beams=1, num_return_sequences=2)
# num_return_sequences > 1, sample # num_return_sequences > 1, sample
self._check_generated_ids(model.generate(input_ids, do_sample=True, num_return_sequences=2)) self._check_generated_ids(model.generate(input_ids, do_sample=True, num_return_sequences=2))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment