From 3367ebd89712239da9773931854d1f1f171ef1a3 Mon Sep 17 00:00:00 2001
From: Christian Tzolov <ctzolov@vmware.com>
Date: Tue, 8 Oct 2024 14:30:08 +0200
Subject: [PATCH] Prevent double-closing of tensors in
 TransformersEmbeddingModel

---
 .../ai/transformers/TransformersEmbeddingModel.java  | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/models/spring-ai-transformers/src/main/java/org/springframework/ai/transformers/TransformersEmbeddingModel.java b/models/spring-ai-transformers/src/main/java/org/springframework/ai/transformers/TransformersEmbeddingModel.java
index f09d8a22e..0619f76c4 100644
--- a/models/spring-ai-transformers/src/main/java/org/springframework/ai/transformers/TransformersEmbeddingModel.java
+++ b/models/spring-ai-transformers/src/main/java/org/springframework/ai/transformers/TransformersEmbeddingModel.java
@@ -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();
+							}
 						}
 					}
 				}