add swift test cases
This commit is contained in:
parent
d3eeeae218
commit
61777707ca
2 changed files with 28 additions and 1 deletions
|
@ -52,7 +52,7 @@ var omniVlmSources = [
|
|||
]
|
||||
|
||||
var testSources = [
|
||||
"tests/LlavaTests/LlavaTests.swift"
|
||||
"swift/LlavaTests/LlavaTests.swift"
|
||||
]
|
||||
|
||||
var resources: [Resource] = []
|
||||
|
|
27
swift/LlavaTests/LlavaTests.swift
Normal file
27
swift/LlavaTests/LlavaTests.swift
Normal file
|
@ -0,0 +1,27 @@
|
|||
import XCTest
|
||||
@testable import omnivlm
|
||||
|
||||
final class LlavaTests: XCTestCase {
|
||||
func testOmniVlm() {
|
||||
omnivlm_init("/Users/liute/Downloads/model-q4_0.gguf",
|
||||
"/Users/liute/Downloads/projector-q4_0.gguf",
|
||||
"vlm-81-instruct")
|
||||
|
||||
let startTime = Date()
|
||||
|
||||
if let cString = omnivlm_inference("describe the image", "/Users/liute/Downloads/cat.png") {
|
||||
let res = String(cString: cString)
|
||||
print("res: \(res)")
|
||||
|
||||
let endTime = Date()
|
||||
let inferenceTime = endTime.timeIntervalSince(startTime)
|
||||
print("Inference time: \(inferenceTime) seconds")
|
||||
|
||||
XCTAssertFalse(res.isEmpty, "res should not be null")
|
||||
} else {
|
||||
XCTFail("failed")
|
||||
}
|
||||
|
||||
omnivlm_free()
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue