Skip to content

Commit 78c6367

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 e1db71c commit 78c6367

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
@@ -382,6 +382,7 @@ def _create_tiny_kokoro_model():
382382
"gpt2": {"model": 44},
383383
"granitemoehybrid": {"model": 118},
384384
"wav2vec2": {"model": 34},
385+
"lasr_ctc": {"model": 384},
385386
"distilbert": {"model": 66},
386387
"t5": {
387388
"encoder": 64,

0 commit comments

Comments
 (0)