Browse Source

fix potentially inaccurate error message

Bruce MacDonald 1 year ago
parent
commit
43a726149d
2 changed files with 2 additions and 2 deletions
  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 (
 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")
 	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" {
 	if runtime.GOOS == "linux" {
 		// check compatibility to log warnings
 		// check compatibility to log warnings
 		if _, err := llm.CheckVRAM(); err != nil {
 		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())
 		}
 		}
 	}
 	}