From 2e7f6a12948dfa93caf7c84d42142e2131560648 Mon Sep 17 00:00:00 2001
From: a252jain <a252jain@uwaterloo.ca>
Date: Fri, 5 Apr 2024 10:51:00 -0400
Subject: [PATCH] name changes
---
Sources/SwiftNLP/2. Encoding/CoreMLEncoder.swift | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/Sources/SwiftNLP/2. Encoding/CoreMLEncoder.swift b/Sources/SwiftNLP/2. Encoding/CoreMLEncoder.swift
index 83988180..bcedf2f7 100644
--- a/Sources/SwiftNLP/2. Encoding/CoreMLEncoder.swift
+++ b/Sources/SwiftNLP/2. Encoding/CoreMLEncoder.swift
@@ -73,7 +73,7 @@ public class MiniLMEmbeddings {
private let model: String
public let tokenizer: BertTokenizer
- public let inputDimention: Int = 128
+ public let inputDimention: Int = 512
public let outputDimention: Int = 384
public init(model_type: String) {
--
GitLab