Skip to content

Commit 309c3bc

Browse files
committed
typo
1 parent 29de539 commit 309c3bc

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

optimum/exporters/openvino/convert.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -322,8 +322,7 @@ def export_pytorch(
322322
logger.info(f"Using framework PyTorch: {torch.__version__}")
323323
output = Path(output)
324324

325-
is_model_stateful = hasattr(model, "use_bettertransformer") and model.use_bettertransformer is True
326-
if stateful and not is_model_stateful:
325+
if stateful:
327326
# Trigger bettertransformer together with stateful model because OpenVINO HW-dependent transformations expect
328327
# both of them are applied to demonstrate the best performance.
329328
# TODO: Consider applying bettertransformer regardless of stateful flag -- requires additional validation.

0 commit comments

Comments
 (0)