From 260565d2c86fd808d7bc183d46383cdad7d69cbd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?F=C3=A9lix=20Marty?=
 <9808326+fxmarty@users.noreply.github.com>
Date: Mon, 16 Oct 2023 16:36:05 +0200
Subject: [PATCH] add details

---
 optimum/exporters/tasks.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
index 242f85ab301..2841383eb96 100644
--- a/optimum/exporters/tasks.py
+++ b/optimum/exporters/tasks.py
@@ -301,8 +301,10 @@ class TasksManager:
             "text-generation-with-past",
             "text2text-generation",
             "text2text-generation-with-past",
-            "text-classification",
-            "question-answering",
+            # text-classification and question-answering can be supported, but the ONNX export is currently broken due to a regression in PyTorch 2.1.
+            # Reference: https://github.com/pytorch/pytorch/issues/110597.
+            # "text-classification",
+            # "question-answering",
             onnx="BartOnnxConfig",
         ),
         # BEiT cannot be used with the masked image modeling autoclass, so this task is excluded here