From e0f939698c0f757973b9aca8816a8b1744d62ec4 Mon Sep 17 00:00:00 2001 From: Jonghwan Hyeon Date: Wed, 1 May 2024 15:03:13 +0000 Subject: [PATCH] Add test for output_attentions --- tests/models/wavlm/test_modeling_wavlm.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/models/wavlm/test_modeling_wavlm.py b/tests/models/wavlm/test_modeling_wavlm.py index c0a8eed2096f35..c9f2f83f7fc0ee 100644 --- a/tests/models/wavlm/test_modeling_wavlm.py +++ b/tests/models/wavlm/test_modeling_wavlm.py @@ -354,6 +354,10 @@ def test_seq_classifier_train(self): config_and_inputs = self.model_tester.prepare_config_and_inputs() self.model_tester.check_seq_classifier_training(*config_and_inputs) + def test_output_attentions(self): + config_and_inputs = self.model_tester.prepare_config_and_inputs() + self.model_tester.check_output_attentions(*config_and_inputs) + def test_labels_out_of_vocab(self): config_and_inputs = self.model_tester.prepare_config_and_inputs() self.model_tester.check_labels_out_of_vocab(*config_and_inputs)