Browse Source

fix potentially inaccurate error message

Bruce MacDonald 1 năm trước cách đây
mục cha
commit
43a726149d
2 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 1 1
      llm/llama.go
  2. 1 1
      server/routes.go

+ 1 - 1
llm/llama.go

@@ -226,7 +226,7 @@ type llama struct {
 }
 
 var (
-	errNvidiaSMI     = errors.New("nvidia-smi command failed")
+	errNvidiaSMI     = errors.New("warning: gpu support may not be enabled, check you have installed GPU drivers: nvidia-smi command failed")
 	errAvailableVRAM = errors.New("not enough VRAM available, falling back to CPU only")
 )
 

+ 1 - 1
server/routes.go

@@ -794,7 +794,7 @@ func Serve(ln net.Listener, allowOrigins []string) error {
 	if runtime.GOOS == "linux" {
 		// check compatibility to log warnings
 		if _, err := llm.CheckVRAM(); err != nil {
-			log.Printf("Warning: GPU support may not be enabled, check you have installed GPU drivers: %v", err)
+			log.Printf(err.Error())
 		}
 	}