diff options
Diffstat (limited to 'model-integration/src/main/java')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java | 7 |
1 files changed, 0 insertions, 7 deletions
diff --git a/model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java b/model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java index d08bc8a3e8c..88028da85c1 100644 --- a/model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java +++ b/model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java @@ -13,8 +13,6 @@ import com.yahoo.tensor.IndexedTensor; import com.yahoo.tensor.Tensor; import com.yahoo.tensor.TensorAddress; import com.yahoo.tensor.TensorType; -import org.slf4j.Logger; -import org.slf4j.LoggerFactory; import java.nio.file.Paths; import java.util.List; @@ -23,8 +21,6 @@ import java.util.Map; @Beta public class HuggingFaceEmbedder extends AbstractComponent implements Embedder { - private static final Logger LOG = LoggerFactory.getLogger(HuggingFaceEmbedder.class.getName()); - private final String inputIdsName; private final String attentionMaskName; private final String tokenTypeIdsName; @@ -147,9 +143,6 @@ public class HuggingFaceEmbedder extends AbstractComponent implements Embedder { return builder.build(); } - private Tensor createAttentionMask(Tensor inputSequence) { - return inputSequence.map((x) -> 1); - } } |