diff --git a/server/images.go b/server/images.go index 75d970db..4a7aa206 100644 --- a/server/images.go +++ b/server/images.go @@ -821,13 +821,11 @@ func downloadBlob(registryURL, repoName, digest string, username, password strin for { fn(fmt.Sprintf("Downloading %s", digest), digest, int(total), int(completed), float64(completed)/float64(total)) if completed >= total { - fmt.Printf("finished downloading\n") - err = os.Rename(fp+"-partial", fp) - if err != nil { - fmt.Printf("error: %v\n", err) + if err := os.Rename(fp+"-partial", fp); err != nil { fn(fmt.Sprintf("error renaming file: %v", err), digest, int(total), int(completed), 1) return err } + break } diff --git a/server/routes.go b/server/routes.go index 67c09948..af7e59fa 100644 --- a/server/routes.go +++ b/server/routes.go @@ -67,8 +67,6 @@ func generate(c *gin.Context) { } req.Prompt = sb.String() - log.Printf("prompt: \n%s", req.Prompt) - llm, err := llama.New(model.ModelPath, opts) if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})