Ver Fonte

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

Jeffrey Morgan há 5 meses atrás
pai
commit
8b4b243f5f
1 ficheiros alterados com 1 adições e 1 exclusões
  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 {