Josh Yan 9 months ago
parent
commit
1344843515
1 changed files with 3 additions and 1 deletions
  1. 3 1
      server/images.go

+ 3 - 1
server/images.go

@@ -440,6 +440,8 @@ func CreateModel(ctx context.Context, name model.Name, modelFileDir, quantizatio
 						defer temp.Close()
 						defer os.Remove(temp.Name())
 
+						// Quantizes per layer
+						// Save total quantized tensors
 						if err := llm.Quantize(blob, temp.Name(), want); err != nil {
 							return err
 						}
@@ -480,7 +482,7 @@ func CreateModel(ctx context.Context, name model.Name, modelFileDir, quantizatio
 			}
 
 			fn(api.ProgressResponse{
-				Status:   fmt.Sprintf("quantizing model %d%%", 100),
+				Status:   "quantizing model done",
 				Quantize: quantization,
 			})
 		case "license", "template", "system":