Skip to content

Commit dc4447c

Browse files
authored
[Model] Fixed the parameter name error in qwen2_audio (#1081)
1 parent 99510de commit dc4447c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

lmms_eval/models/simple/qwen2_audio.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ def _collate(x):
233233
else:
234234
text = ["<|audio_bos|><|AUDIO|><|audio_eos|>" + context for context in contexts]
235235

236-
inputs = self.processor(text=text, audios=audios, return_tensors="pt", padding=True, sampling_rate=self.processor.feature_extractor.sampling_rate)
236+
inputs = self.processor(text=text, audio=audios, return_tensors="pt", padding=True, sampling_rate=self.processor.feature_extractor.sampling_rate)
237237

238238
if self.device_map == "auto":
239239
inputs = inputs.to("cuda")

0 commit comments

Comments
 (0)