This commit is contained in:
Awni Hannun
2023-12-14 16:56:50 -08:00
parent e434e7e5c2
commit b863e7cca0
8 changed files with 33 additions and 11 deletions

View File

@@ -65,7 +65,6 @@ class TestWhisper(unittest.TestCase):
logits = mlx_model(mels, tokens)
self.assertEqual(logits.dtype, mx.float16)
def test_decode_lang(self):
options = decoding.DecodingOptions(task="lang_id", fp16=False)
result = decoding.decode(self.model, self.mels, options)