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