update test case

This commit is contained in:
Te993 2025-01-10 13:56:29 +08:00
parent 61777707ca
commit 91ab9ed858

View file

@ -3,13 +3,13 @@ import XCTest
final class LlavaTests: XCTestCase { final class LlavaTests: XCTestCase {
func testOmniVlm() { func testOmniVlm() {
omnivlm_init("/Users/liute/Downloads/model-q4_0.gguf", omnivlm_init("model-q4_0.gguf",
"/Users/liute/Downloads/projector-q4_0.gguf", "projector-q4_0.gguf",
"vlm-81-instruct") "vlm-81-instruct")
let startTime = Date() let startTime = Date()
if let cString = omnivlm_inference("describe the image", "/Users/liute/Downloads/cat.png") { if let cString = omnivlm_inference("describe the image", "cat.png") {
let res = String(cString: cString) let res = String(cString: cString)
print("res: \(res)") print("res: \(res)")