remove types

This commit is contained in:
Josh Yan 2024-08-12 09:47:05 -07:00
parent 9bc42f532b
commit 73d69bc90b
2 changed files with 2 additions and 5 deletions

View File

@ -65,13 +65,11 @@ 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 %d/%d", int(progress), tensorCount),
Type: "quantize",
Status: fmt.Sprintf("quantizing model %d%%", 100*int(progress)/tensorCount),
})
case <-done:
fn(api.ProgressResponse{
Status: fmt.Sprintf("quantizing model %d/%d", tensorCount, tensorCount),
Type: "quantize",
Status: fmt.Sprintf("quantizing model 100%%"),
})
return
}

View File

@ -442,7 +442,6 @@ func CreateModel(ctx context.Context, name model.Name, modelFileDir, quantizatio
} else if want != ft {
fn(api.ProgressResponse{
Status: "quantizing model tensors",
Type: "quantize",
})
blob, err := GetBlobsPath(baseLayer.Digest)