This commit is contained in:
Josh Yan 2024-08-01 16:49:03 -07:00
parent 4a7bfca902
commit 07c0f66f5e

View File

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