Forráskód Böngészése

only quantize language models

Michael Yang 1 éve
szülő
commit
05105903d8
1 módosított fájl, 4 hozzáadás és 2 törlés
  1. 4 2
      server/images.go

+ 4 - 2
server/images.go

@@ -376,7 +376,10 @@ func CreateModel(ctx context.Context, name, modelFileDir, quantization string, m
 			}
 			}
 
 
 			for _, baseLayer := range baseLayers {
 			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)
 					ftype, err := llm.ParseFileType(quantization)
 					if err != nil {
 					if err != nil {
 						return err
 						return err
@@ -582,7 +585,6 @@ func CreateModel(ctx context.Context, name, modelFileDir, quantization string, m
 	return nil
 	return nil
 }
 }
 
 
-
 func CopyModel(src, dst model.Name) error {
 func CopyModel(src, dst model.Name) error {
 	if !dst.IsFullyQualified() {
 	if !dst.IsFullyQualified() {
 		return model.Unqualified(dst)
 		return model.Unqualified(dst)