Skip to content

Commit 88b01fd

Browse files
Update expected output IDs in XGLM generation test
1 parent d602952 commit 88b01fd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/xglm/test_modeling_xglm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -357,7 +357,7 @@ def _test_lm_generate_xglm_helper(
357357
model.to(torch_device)
358358
input_ids = torch.tensor([[2, 268, 9865]], dtype=torch.long, device=torch_device) # The dog
359359
# </s> The dog is a very friendly dog. He is very affectionate and loves to play with other
360-
expected_output_ids = [2, 268, 9865, 67, 11, 1988, 57252, 9865, 5, 984, 67, 1988, 213838, 1658, 53, 70446, 33, 6657, 278, 1581] # fmt: skip
360+
expected_output_ids = [2, 268, 9865, 67, 11, 1988, 57252, 9865, 5, 984, 67, 1988, 213838, 1658, 53, 70446, 33, 6657, 278, 1581, 72616, 5, 984] # fmt: skip
361361
output_ids = model.generate(input_ids, do_sample=False, num_beams=1)
362362
if verify_outputs:
363363
self.assertListEqual(output_ids[0].tolist(), expected_output_ids)

0 commit comments

Comments
 (0)