diff --git a/integration/llm_image_test.go b/integration/llm_image_test.go index b9726c8f..bbd031a9 100644 --- a/integration/llm_image_test.go +++ b/integration/llm_image_test.go @@ -19,7 +19,7 @@ func TestVisionModels(t *testing.T) { } testCases := []testCase{ { - model: "llava:7b", + model: "qwen2.5vl", }, { model: "llama3.2-vision", @@ -60,6 +60,7 @@ func TestVisionModels(t *testing.T) { } func TestIntegrationSplitBatch(t *testing.T) { + skipUnderMinVRAM(t, 6) image, err := base64.StdEncoding.DecodeString(imageEncoding) require.NoError(t, err) req := api.GenerateRequest{