retry download on network errors

This commit is contained in:
Bruce MacDonald
2023-08-15 15:07:19 -03:00
parent 12052a7624
commit f0d7c2f5ea
2 changed files with 51 additions and 22 deletions

View File

@@ -952,7 +952,14 @@ func PullModel(ctx context.Context, name string, regOpts *RegistryOptions, fn fu
layers = append(layers, &manifest.Config)
for _, layer := range layers {
if err := downloadBlob(ctx, mp, layer.Digest, regOpts, fn); err != nil {
if err := downloadBlob(
ctx,
downloadOpts{
mp: mp,
digest: layer.Digest,
regOpts: regOpts,
fn: fn,
}); err != nil {
return err
}
}