• Stas Bekman's avatar
    [testing] fix ambiguous test (#6898) · e71f32c0
    Stas Bekman authored
    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.
    e71f32c0
test_modeling_common.py 47.2 KB