We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2588077 commit e4a2091Copy full SHA for e4a2091
optimum/exporters/openvino/convert.py
@@ -299,7 +299,7 @@ def export_pytorch(
299
logger.info(f"Using framework PyTorch: {torch.__version__}")
300
output = Path(output)
301
302
- if stateful:
+ if ensure_export_task_support_stateful(config.task):
303
# Trigger bettertransformer together with stateful model because OpenVINO HW-dependent transformations expect
304
# both of them are applied to demonstrate the best performance.
305
# TODO: Consider applying bettertransformer regardless of stateful flag -- requires additional validation.
0 commit comments