Skip to content
Snippets Groups Projects
Commit 2e7f6a12 authored by Abhinav Jain's avatar Abhinav Jain
Browse files

name changes

parent 8756de95
No related branches found
No related tags found
1 merge request!15Add interface for using generic CoreML LLMs
Pipeline #116028 passed with warnings
Stage: build
Stage: test
Hide whitespace changes
Inline Side-by-side
Sources/SwiftNLP/2. Encoding/CoreMLEncoder.swift
+ 1
− 1
View file @ 2e7f6a12
...@@ -73,7 +73,7 @@ public class MiniLMEmbeddings { ...@@ -73,7 +73,7 @@ public class MiniLMEmbeddings {
private let model: String private let model: String
public let tokenizer: BertTokenizer public let tokenizer: BertTokenizer
public let inputDimention: Int = 128 public let inputDimention: Int = 512
public let outputDimention: Int = 384 public let outputDimention: Int = 384
public init(model_type: String) { public init(model_type: String) {
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment