Update optimum/exporters/tasks.py

Co-authored-by: Ella Charlaix <[email protected]>
huggingface · Jan 7, 2025 · f019033 · f019033
1 parent b26efad
commit f019033
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
@@ -314,9 +314,7 @@ class TasksManager:
     }
 
     _CUSTOM_CLASSES = {
-        ("pt", "patchtsmixer", "feature-extraction"): ("transformers", "PatchTSMixerModel"),
         ("pt", "patchtsmixer", "time-series-forecasting"): ("transformers", "PatchTSMixerForPrediction"),
-        ("pt", "patchtst", "feature-extraction"): ("transformers", "PatchTSTModel"),
         ("pt", "patchtst", "time-series-forecasting"): ("transformers", "PatchTSTForPrediction"),
         ("pt", "pix2struct", "image-to-text"): ("transformers", "Pix2StructForConditionalGeneration"),
         ("pt", "pix2struct", "visual-question-answering"): ("transformers", "Pix2StructForConditionalGeneration"),