Merge pull request ollama#1347 from jshph/adapter-hash

Fix adapter loading from SHA hash
Eastwooder · Dec 1, 2023 · cedae0d · cedae0d
2 parents 6681d37 + bb80a59
commit cedae0d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/server/images.go b/server/images.go
@@ -375,6 +375,15 @@ func CreateModel(ctx context.Context, name, modelFileDir string, commands []pars
 			layer.MediaType = mediatype
 			layers = append(layers, layer)
 		case "adapter":
+			if strings.HasPrefix(c.Args, "@") {
+				blobPath, err := GetBlobsPath(strings.TrimPrefix(c.Args, "@"))
+				if err != nil {
+					return err
+				}
+
+				c.Args = blobPath
+			}
+
 			fn(api.ProgressResponse{Status: "creating adapter layer"})
 			bin, err := os.Open(realpath(modelFileDir, c.Args))
 			if err != nil {