Skip to content

Commit 36e5731

Browse files
nikita-savelyevvmvafin
authored andcommitted
Update reference (#1386)
1 parent f485773 commit 36e5731

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/openvino/utils_tests.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
import torch
2121

2222
from optimum.intel.openvino.modeling_base import OVBaseModel
23-
from optimum.intel.utils.import_utils import is_openvino_version
23+
from optimum.intel.utils.import_utils import is_nncf_version, is_openvino_version
2424

2525

2626
MODEL_NAMES = {
@@ -301,7 +301,7 @@
301301
"transformer": 58,
302302
"vae_decoder": 28,
303303
"vae_encoder": 28,
304-
"text_encoder": 18,
304+
"text_encoder": 18 if is_nncf_version("<", "2.18.0") else 16,
305305
},
306306
"ltx-video": {
307307
"transformer": 34,

0 commit comments

Comments
 (0)