瀏覽代碼

only quantize language models

Michael Yang 1 年之前
父節點
當前提交
d245460362
共有 1 個文件被更改,包括 4 次插入2 次删除
  1. 4 2
      server/images.go

+ 4 - 2
server/images.go

@@ -377,7 +377,10 @@ func CreateModel(ctx context.Context, name, modelFileDir, quantization string, m
 			}
 
 			for _, baseLayer := range baseLayers {
-				if quantization != "" && baseLayer.GGML != nil && baseLayer.GGML.Name() == "gguf" {
+				if quantization != "" &&
+					baseLayer.MediaType == "application/vnd.ollama.image.model" &&
+					baseLayer.GGML != nil &&
+					baseLayer.GGML.Name() == "gguf" {
 					ftype, err := llm.ParseFileType(quantization)
 					if err != nil {
 						return err
@@ -582,7 +585,6 @@ func CreateModel(ctx context.Context, name, modelFileDir, quantization string, m
 	return nil
 }
 
-
 func CopyModel(src, dst model.Name) error {
 	if !dst.IsFullyQualified() {
 		return model.Unqualified(dst)