in src/exporters/coreml/config.py [0:0]
def fill_outputs_with_past_key_values_(self, outputs: "OrderedDict[str, OutputDescription]"):
# TODO: Temporarily disabled until we can solve the issue with encoder past key/values
# name = "decoder_present" if self.seq2seq == "decoder" else "present"
name = "present"
for i in range(self.num_layers):
outputs[f"{name}_{i}_key"] = OutputDescription(f"{name}_{i}_key")
outputs[f"{name}_{i}_value"] = OutputDescription(f"{name}_{i}_value")