Skip to content

Commit 0d8e57c

Browse files
committed
fix(tests): transformer-musicgen tests fixups
1 parent 8593817 commit 0d8e57c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backend/python/transformers-musicgen/test_transformers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def test_tts(self):
7171
stub = backend_pb2_grpc.BackendStub(channel)
7272
response = stub.LoadModel(backend_pb2.ModelOptions(Model="facebook/musicgen-small"))
7373
self.assertTrue(response.success)
74-
tts_request = backend_pb2.TTSRequest(Model="facebook/musicgen-small", Input="80s TV news production music hit for tonight's biggest story")
74+
tts_request = backend_pb2.TTSRequest(text="80s TV news production music hit for tonight's biggest story")
7575
tts_response = stub.TTS(tts_request)
7676
self.assertIsNotNone(tts_response)
7777
except Exception as err:

0 commit comments

Comments
 (0)