From 1071e176260857c579fbb140e95fc31d793b4036 Mon Sep 17 00:00:00 2001 From: Roy Han Date: Fri, 28 Jun 2024 13:14:49 -0700 Subject: [PATCH] lint --- cmd/cmd.go | 4 ++-- server/routes.go | 2 -- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/cmd/cmd.go b/cmd/cmd.go index debb39218..7881399a1 100644 --- a/cmd/cmd.go +++ b/cmd/cmd.go @@ -526,7 +526,7 @@ func ListRunningHandler(cmd *cobra.Command, args []string) error { var data [][]string for _, m := range models.Models { - if len(args) == 0 || strings.HasPrefix(m.Name, args[0]) { + if len(args) == 0 || strings.HasPrefix(m.Model, args[0]) { var procStr string switch { case m.SizeVRAM == 0: @@ -540,7 +540,7 @@ func ListRunningHandler(cmd *cobra.Command, args []string) error { cpuPercent := math.Round(float64(sizeCPU) / float64(m.Size) * 100) procStr = fmt.Sprintf("%d%%/%d%% CPU/GPU", int(cpuPercent), int(100-cpuPercent)) } - data = append(data, []string{m.Name, m.Digest[:12], format.HumanBytes(m.Size), procStr, format.HumanTime(m.ExpiresAt, "Never")}) + data = append(data, []string{m.Model, m.Digest[:12], format.HumanBytes(m.Size), procStr, format.HumanTime(m.ExpiresAt, "Never")}) } } diff --git a/server/routes.go b/server/routes.go index ff66663c0..1db1b86dc 100644 --- a/server/routes.go +++ b/server/routes.go @@ -800,7 +800,6 @@ func (s *Server) ListModelsHandler(c *gin.Context) { // tag should never be masked models = append(models, api.ListModelResponse{ - Model: n.DisplayShortest(), Name: n.DisplayShortest(), Size: m.Size(), Digest: m.digest, @@ -1219,7 +1218,6 @@ func (s *Server) ProcessHandler(c *gin.Context) { mr := api.ProcessModelResponse{ Model: model.ShortName, - Name: model.ShortName, Size: int64(v.estimatedTotal), SizeVRAM: int64(v.estimatedVRAM), Digest: model.Digest,