Browse Source

fix: quantization with template

Michael Yang 10 months ago
parent
commit
e835ef1836
1 changed files with 10 additions and 5 deletions
  1. 10 5
      server/images.go

+ 10 - 5
server/images.go

@@ -414,17 +414,22 @@ func CreateModel(ctx context.Context, name model.Name, modelFileDir, quantizatio
 							return err
 						}
 
-						layers, err := parseFromFile(ctx, temp, "", fn)
+						layer, err := NewLayer(temp, baseLayer.MediaType)
 						if err != nil {
 							return err
 						}
 
-						if len(layers) != 1 {
-							return errors.New("quantization failed")
+						if _, err := temp.Seek(0, io.SeekStart); err != nil {
+							return err
+						}
+
+						ggml, _, err := llm.DecodeGGML(temp)
+						if err != nil {
+							return err
 						}
 
-						baseLayer.Layer = layers[0].Layer
-						baseLayer.GGML = layers[0].GGML
+						baseLayer.Layer = layer
+						baseLayer.GGML = ggml
 					}
 				}