Skip to content

GitLab

Unverified Commit aed636f5 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[tests] fix clapconfig for text backbone in audioldm2 (#12490) 

fix clapconfig for text backbone in audioldm2
parent 53a10518
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
tests/pipelines/audioldm2/test_audioldm2.py
View file @ aed636f5
...@@ -138,10 +138,8 @@ class AudioLDM2PipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -138,10 +138,8 @@ class AudioLDM2PipelineFastTests(PipelineTesterMixin, unittest.TestCase):
patch_stride=2, patch_stride=2,
patch_embed_input_channels=4, patch_embed_input_channels=4,
) )
text_encoder_config = ClapConfig.from_text_audio_configs( text_encoder_config = ClapConfig(
text_config=text_branch_config, text_config=text_branch_config, audio_config=audio_branch_config, projection_dim=16
audio_config=audio_branch_config,
projection_dim=16,
) )
text_encoder = ClapModel(text_encoder_config) text_encoder = ClapModel(text_encoder_config)
tokenizer = RobertaTokenizer.from_pretrained("hf-internal-testing/tiny-random-roberta", model_max_length=77) tokenizer = RobertaTokenizer.from_pretrained("hf-internal-testing/tiny-random-roberta", model_max_length=77)
......
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