瀏覽代碼

server: fix warnings in prompt_test.go (#7710)

Jeffrey Morgan 5 月之前
父節點
當前提交
8b4b243f5f
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      server/prompt_test.go

+ 1 - 1
server/prompt_test.go

@@ -32,7 +32,7 @@ func TestChatPrompt(t *testing.T) {
 	mllamaModel := Model{Template: tmpl, ProjectorPaths: []string{"vision"}, Config: ConfigV2{ModelFamilies: []string{"mllama"}}}
 
 	createImg := func(width, height int) ([]byte, error) {
-		img := image.NewRGBA(image.Rect(0, 0, 5, 5))
+		img := image.NewRGBA(image.Rect(0, 0, width, height))
 		var buf bytes.Buffer
 
 		if err := png.Encode(&buf, img); err != nil {