From d211a84aca8a8513a599171173c29bc8d3061fba Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 5 Jul 2023 11:37:46 +0200 Subject: [PATCH] Fix `EncodecModelTest::test_multi_gpu_data_parallel_forward` (#24663) fix Co-authored-by: ydshieh --- tests/models/encodec/test_modeling_encodec.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/encodec/test_modeling_encodec.py b/tests/models/encodec/test_modeling_encodec.py index d678b707a5..389444d4b2 100644 --- a/tests/models/encodec/test_modeling_encodec.py +++ b/tests/models/encodec/test_modeling_encodec.py @@ -73,7 +73,8 @@ class EncodecModelTester: def __init__( self, parent, - batch_size=13, + # `batch_size` needs to be an even number if the model has some outputs with batch dim != 0. + batch_size=12, num_channels=2, is_training=False, num_hidden_layers=4,