Prevent double-closing of tensors in TransformersEmbeddingModel

This commit is contained in:
Christian Tzolov
2024-10-08 14:30:08 +02:00
parent e30105e5fb
commit 3367ebd897

View File

@@ -341,9 +341,15 @@ public class TransformersEmbeddingModel extends AbstractEmbeddingModel implement
}
}
finally {
inputIds.close();
attentionMask.close();
tokenTypeIds.close();
if (!inputIds.isClosed()) {
inputIds.close();
}
if (!attentionMask.isClosed()) {
attentionMask.close();
}
if (!tokenTypeIds.isClosed()) {
tokenTypeIds.close();
}
}
}
}