highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-07-28 18:15:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

progress

Browse Source
This commit is contained in:
Michael Yang
2023-07-06 14:05:55 -07:00
parent 3d6009aae3
commit c4b9e84945
6 changed files with 75 additions and 152 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
server/models.go
View File

@@ -78,7 +78,7 @@ func saveModel(model *Model, progressCh chan<- api.PullProgress) error {
return fmt.Errorf("failed to download model: %w", err)
}
// check for resume
alreadyDownloaded := 0
alreadyDownloaded := int64(0)
fileInfo, err := os.Stat(fileName)
if err != nil {
if !os.IsNotExist(err) {
@@ -86,8 +86,8 @@ func saveModel(model *Model, progressCh chan<- api.PullProgress) error {
}
// file doesn't exist, create it now
} else {
alreadyDownloaded = int(fileInfo.Size())
req.Header.Add("Range", "bytes="+strconv.Itoa(alreadyDownloaded)+"-")
alreadyDownloaded = fileInfo.Size()
req.Header.Add("Range", fmt.Sprintf("bytes=%d-", alreadyDownloaded))
}
resp, err := client.Do(req)
@@ -117,7 +117,7 @@ func saveModel(model *Model, progressCh chan<- api.PullProgress) error {
}
defer out.Close()
totalSize, _ := strconv.Atoi(resp.Header.Get("Content-Length"))
totalSize, _ := strconv.ParseInt(resp.Header.Get("Content-Length"), 10, 64)
buf := make([]byte, 1024)
totalBytes := alreadyDownloaded
@@ -134,7 +134,8 @@ func saveModel(model *Model, progressCh chan<- api.PullProgress) error {
if _, err := out.Write(buf[:n]); err != nil {
return err
}
totalBytes += n
totalBytes += int64(n)
// send progress updates
progressCh <- api.PullProgress{