SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix multi_gpu_data_parallel_forward for MusicgenTest (#29632)

Browse Source 
update

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-03-13 19:12:20 +01:00
committed by GitHub
parent 5ac264d8a8
commit fe085560d0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/musicgen/test_modeling_musicgen.py
View File

@@ -103,7 +103,7 @@ class MusicgenDecoderTester:
def __init__(
self,
parent,
batch_size=3, # need batch_size != num_hidden_layers
batch_size=4, # need batch_size != num_hidden_layers
seq_length=7,
is_training=False,
use_labels=False,
@@ -441,7 +441,7 @@ class MusicgenTester:
def __init__(
self,
parent,
batch_size=3, # need batch_size != num_hidden_layers
batch_size=4, # need batch_size != num_hidden_layers
seq_length=7,
is_training=False,
use_labels=False,