Skip to content

Commit 7ba04dc

Browse files
committed
Fix quantization test for lasr_ctc (MedASR)
Add expected INT8 weight count for lasr_ctc model in _ARCHITECTURES_TO_EXPECTED_INT8 to prevent KeyError during test_ovmodel_load_with_compressed_weights.
1 parent edcd12c commit 7ba04dc

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/openvino/utils_tests.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -371,6 +371,7 @@ def _create_tiny_kokoro_model():
371371
"gpt2": {"model": 44},
372372
"granitemoehybrid": {"model": 118},
373373
"wav2vec2": {"model": 34},
374+
"lasr_ctc": {"model": 384},
374375
"distilbert": {"model": 66},
375376
"t5": {
376377
"encoder": 64,

0 commit comments

Comments
 (0)