highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-08-28 13:59:53 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
return on part done
This commit is contained in:
Michael Yang
2024-05-15 17:16:24 -07:00
committed by GitHub
parent 7e9a2da097 ac145f75ca
commit 84ed77cbd8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server/download.go
View File

@@ -221,7 +221,7 @@ func (b *blobDownload) downloadChunk(ctx context.Context, requestURL *url.URL, w
}
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) {
// rollback progress
b.Completed.Add(-n)