Browse Source

build release mode

Michael Yang 1 year ago
parent
commit
95187d7e1e
2 changed files with 15 additions and 0 deletions
  1. 1 0
      scripts/build_darwin.sh
  2. 14 0
      server/routes.go

+ 1 - 0
scripts/build_darwin.sh

@@ -3,6 +3,7 @@
 mkdir -p dist
 
 GO_LDFLAGS="-X github.com/jmorganca/ollama/version.Version=$VERSION"
+GO_LDFLAGS="$GO_LDFLAGS -X github.com/jmorganca/ollama/server.mode=release"
 
 # build universal binary
 CGO_ENABLED=1 GOARCH=arm64 go build -ldflags "$GO_LDFLAGS" -o dist/ollama-darwin-arm64

+ 14 - 0
server/routes.go

@@ -25,6 +25,20 @@ import (
 	"github.com/jmorganca/ollama/vector"
 )
 
+var mode string = gin.DebugMode
+
+func init() {
+	switch mode {
+	case gin.DebugMode:
+	case gin.ReleaseMode:
+	case gin.TestMode:
+	default:
+		mode = gin.DebugMode
+	}
+
+	gin.SetMode(mode)
+}
+
 var loaded struct {
 	mu sync.Mutex