change progress msg

This commit is contained in:
Josh Yan 2024-07-31 10:57:25 -07:00
parent 04f2154505
commit 4a7bfca902

View File

@ -65,13 +65,13 @@ func Quantize(infile, outfile string, ftype fileType, fn func(resp api.ProgressR
progressInt := atomic.LoadInt32(store)
progress := *(*float32)(unsafe.Pointer(&progressInt))
fn(api.ProgressResponse{
Status: fmt.Sprintf("quantizing model tensors %d/%d", int(progress), tensorCount),
Status: fmt.Sprintf("quantizing model %d/%d", int(progress), tensorCount),
Type: "quantize",
})
fmt.Println("Progress: ", progress)
case <-done:
fn(api.ProgressResponse{
Status: fmt.Sprintf("quantizing model tensors %d/%d", tensorCount, tensorCount),
Status: fmt.Sprintf("quantizing model %d/%d", tensorCount, tensorCount),
Type: "quantize",
})
return