remove debugging messages

This commit is contained in:
Michael Yang 2023-07-17 12:39:55 -07:00
parent 53d0052c6c
commit c7dd52271c
2 changed files with 2 additions and 6 deletions

View file

@ -821,13 +821,11 @@ func downloadBlob(registryURL, repoName, digest string, username, password strin
for { for {
fn(fmt.Sprintf("Downloading %s", digest), digest, int(total), int(completed), float64(completed)/float64(total)) fn(fmt.Sprintf("Downloading %s", digest), digest, int(total), int(completed), float64(completed)/float64(total))
if completed >= total { if completed >= total {
fmt.Printf("finished downloading\n") if err := os.Rename(fp+"-partial", fp); err != nil {
err = os.Rename(fp+"-partial", fp)
if err != nil {
fmt.Printf("error: %v\n", err)
fn(fmt.Sprintf("error renaming file: %v", err), digest, int(total), int(completed), 1) fn(fmt.Sprintf("error renaming file: %v", err), digest, int(total), int(completed), 1)
return err return err
} }
break break
} }

View file

@ -67,8 +67,6 @@ func generate(c *gin.Context) {
} }
req.Prompt = sb.String() req.Prompt = sb.String()
log.Printf("prompt: \n%s", req.Prompt)
llm, err := llama.New(model.ModelPath, opts) llm, err := llama.New(model.ModelPath, opts)
if err != nil { if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})