Skip to content

GitLab

Unverified Commit 84ed77cb authored by Michael Yang's avatar Michael Yang Committed by GitHub
Browse files

Merge pull request #4436 from ollama/mxyng/done-part 

return on part done
parents 7e9a2da0 ac145f75
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
server/download.go
View file @ 84ed77cb
...@@ -221,7 +221,7 @@ func (b *blobDownload) downloadChunk(ctx context.Context, requestURL *url.URL, w ...@@ -221,7 +221,7 @@ func (b *blobDownload) downloadChunk(ctx context.Context, requestURL *url.URL, w
} }
defer resp.Body.Close() defer resp.Body.Close()
n, err := io.Copy(w, io.TeeReader(resp.Body, part)) n, err := io.CopyN(w, io.TeeReader(resp.Body, part), part.Size)
if err != nil && !errors.Is(err, context.Canceled) && !errors.Is(err, io.ErrUnexpectedEOF) { if err != nil && !errors.Is(err, context.Canceled) && !errors.Is(err, io.ErrUnexpectedEOF) {
// rollback progress // rollback progress
b.Completed.Add(-n) b.Completed.Add(-n)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment