Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "9af845afc2607172b6830610ab465bdd31f258cd"
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: ...@@ -103,7 +103,7 @@ class MusicgenDecoderTester:
def __init__( def __init__(
self, self,
parent, parent,
batch_size=3, # need batch_size != num_hidden_layers batch_size=4, # need batch_size != num_hidden_layers
seq_length=7, seq_length=7,
is_training=False, is_training=False,
use_labels=False, use_labels=False,
...@@ -441,7 +441,7 @@ class MusicgenTester: ...@@ -441,7 +441,7 @@ class MusicgenTester:
def __init__( def __init__(
self, self,
parent, parent,
batch_size=3, # need batch_size != num_hidden_layers batch_size=4, # need batch_size != num_hidden_layers
seq_length=7, seq_length=7,
is_training=False, is_training=False,
use_labels=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