Skip to content

Commit aed636f

Browse files
authored
[tests] fix clapconfig for text backbone in audioldm2 (huggingface#12490)
fix clapconfig for text backbone in audioldm2
1 parent 53a1051 commit aed636f

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/pipelines/audioldm2/test_audioldm2.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -138,10 +138,8 @@ def get_dummy_components(self):
138138
patch_stride=2,
139139
patch_embed_input_channels=4,
140140
)
141-
text_encoder_config = ClapConfig.from_text_audio_configs(
142-
text_config=text_branch_config,
143-
audio_config=audio_branch_config,
144-
projection_dim=16,
141+
text_encoder_config = ClapConfig(
142+
text_config=text_branch_config, audio_config=audio_branch_config, projection_dim=16
145143
)
146144
text_encoder = ClapModel(text_encoder_config)
147145
tokenizer = RobertaTokenizer.from_pretrained("hf-internal-testing/tiny-random-roberta", model_max_length=77)

0 commit comments

Comments
 (0)