"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "eac5b8bfbdd1f1bd949b7dce2793b5afc5f64d02"
[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.
Showing
Please register or sign in to comment