Jelajahi Sumber

only quantize language models

Michael Yang 1 tahun lalu
induk
melakukan
3244a25c79
1 mengubah file dengan 4 tambahan dan 2 penghapusan
  1. 4 2
      server/images.go

+ 4 - 2
server/images.go

@@ -349,7 +349,10 @@ func CreateModel(ctx context.Context, name, modelFileDir, quantization string, c
 			}
 
 			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
@@ -555,7 +558,6 @@ func CreateModel(ctx context.Context, name, modelFileDir, quantization string, c
 	return nil
 }
 
-
 func CopyModel(src, dst model.Name) error {
 	if !dst.IsFullyQualified() {
 		return model.Unqualified(dst)