gered/ollama
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

clean up model pull

Browse source
This commit is contained in:
Bruce MacDonald 2023-07-06 15:03:52 -04:00 committed by Jeffrey Morgan
parent 350aec772b
commit d436d51c78
1 changed files with 1 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
server/models.go
View file

@ -35,7 +35,6 @@ func pull(model string, progressCh chan<- api.PullProgress) error {
if err != nil { if err != nil {
return fmt.Errorf("failed to pull model: %w", err) return fmt.Errorf("failed to pull model: %w", err)
} }
return saveModel(remote, progressCh) return saveModel(remote, progressCh)
} }
@ -76,7 +75,7 @@ func saveModel(model *Model, progressCh chan<- api.PullProgress) error {
client := &http.Client{} client := &http.Client{}
req, err := http.NewRequest("GET", model.URL, nil) req, err := http.NewRequest("GET", model.URL, nil)
if err != nil { if err != nil {
panic(err) return fmt.Errorf("failed to download model: %w", err)
} }
// check for resume // check for resume
alreadyDownloaded := 0 alreadyDownloaded := 0
@ -126,19 +125,15 @@ func saveModel(model *Model, progressCh chan<- api.PullProgress) error {
for { for {
n, err := resp.Body.Read(buf) n, err := resp.Body.Read(buf)
if err != nil && err != io.EOF { if err != nil && err != io.EOF {
return err return err
} }
if n == 0 { if n == 0 {
break break
} }
if _, err := out.Write(buf[:n]); err != nil { if _, err := out.Write(buf[:n]); err != nil {
return err return err
} }
totalBytes += n totalBytes += n
// send progress updates // send progress updates