Fix distilbert export

mfuntowicz · mfuntowicz · commit 6b1899ffbc55 · 2021-06-28T20:08:09.000+02:00
diff --git a/src/transformers/models/distilbert/configuration_distilbert.py b/src/transformers/models/distilbert/configuration_distilbert.py
@@ -139,7 +139,6 @@ def num_hidden_layers(self):
         return self.n_layers
 
 
-# Copied from transformers.models.bert.configuration_bert.BertOnnxConfig with Roberta->DistilBert
 class DistilBertOnnxConfig(OnnxConfig):
     @property
     def inputs(self) -> Mapping[str, Mapping[int, str]]:
@@ -150,5 +149,5 @@ def inputs(self) -> Mapping[str, Mapping[int, str]]:
 
     @property
     def outputs(self) -> Mapping[str, Mapping[int, str]]:
-        return {"last_hidden_state": {0: "batch", 1: "sequence"}, "pooler_output": {0: "batch"}}
+        return {"last_hidden_state": {0: "batch", 1: "sequence"}}
 
