diff --git a/Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift b/Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift index c44b98da1aa4949cb15cfd6eb88de7dad84a7fa0..d1e5e35b6d3af972b2dbeffb156d7a530ef5c7ce 100644 --- a/Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift +++ b/Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift @@ -34,7 +34,7 @@ final class BERT_test: XCTestCase { var query_embedding: [Float] = [] 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])! @@ -46,7 +46,7 @@ final class BERT_test: XCTestCase { //print(i) i += 1 } else { - fatalError("Error occurred1") + fatalError("Error occurred!") } }