Skip to content

GitLab

Unverified Commit fe085560 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `multi_gpu_data_parallel_forward` for `MusicgenTest` (#29632) 



update
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 5ac264d8
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/musicgen/test_modeling_musicgen.py
View file @ fe085560
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment