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

name changes

parent 64c754af
No related branches found
No related tags found
1 merge request!15Add interface for using generic CoreML LLMs
Pipeline #116020 failed
Stage: build
Stage: test
Hide whitespace changes
Inline Side-by-side
Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift
+ 2
− 2
View file @ 6ea88ef5
...@@ -34,7 +34,7 @@ final class BERT_test: XCTestCase { ...@@ -34,7 +34,7 @@ final class BERT_test: XCTestCase {
var query_embedding: [Float] = [] var query_embedding: [Float] = []
var embedding_dim: Int = 384 var embedding_dim: Int = 384
var model = MiniLMEmbeddings(model_type: "all_miniLM_L6_v2") var model = MiniLMEmbeddings(model_type: "all_MiniLM_L6_v2")
query_embedding = await model.encode(sentence: query[0])! query_embedding = await model.encode(sentence: query[0])!
...@@ -46,7 +46,7 @@ final class BERT_test: XCTestCase { ...@@ -46,7 +46,7 @@ final class BERT_test: XCTestCase {
//print(i) //print(i)
i += 1 i += 1
} else { } else {
fatalError("Error occurred1") fatalError("Error occurred!")
} }
} }
......
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