Browse Source

Merge pull request #1347 from jshph/adapter-hash

Fix adapter loading from SHA hash
Michael Yang 1 year ago
parent
commit
cedae0d17a
1 changed files with 9 additions and 0 deletions
  1. 9 0
      server/images.go

+ 9 - 0
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 {