Merge pull request #82 from jmorganca/filepath

windows build
This commit is contained in:
Michael Yang 2023-07-14 20:11:55 -07:00 committed by GitHub
commit 6c9fa573ae
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 15 additions and 15 deletions

View file

@ -9,7 +9,7 @@ import (
"net"
"net/http"
"os"
"path"
"path/filepath"
"strings"
"time"
@ -27,7 +27,7 @@ func cacheDir() string {
panic(err)
}
return path.Join(home, ".ollama")
return filepath.Join(home, ".ollama")
}
func RunRun(cmd *cobra.Command, args []string) error {
@ -209,7 +209,7 @@ func NewCLI() *cobra.Command {
},
PersistentPreRunE: func(_ *cobra.Command, args []string) error {
// create the models directory and it's parent
return os.MkdirAll(path.Join(cacheDir(), "models"), 0o700)
return os.MkdirAll(filepath.Join(cacheDir(), "models"), 0o700)
},
}

View file

@ -264,8 +264,8 @@ func (llm *llama) sample(output deque[C.llama_token], opts *C.struct_llama_sampl
token := C.llama_sample(
llm.ctx,
unsafe.SliceData(candidates.Data()), C.ulong(candidates.Len()),
unsafe.SliceData(output.Data()), C.ulong(output.Len()),
unsafe.SliceData(candidates.Data()), C.size_t(candidates.Len()),
unsafe.SliceData(output.Data()), C.size_t(output.Len()),
opts)
if token != C.llama_token_eos() {
return token, nil

View file

@ -7,7 +7,7 @@ import (
"io"
"net/http"
"os"
"path"
"path/filepath"
"strconv"
)
@ -32,14 +32,14 @@ func (m *Model) FullName() string {
panic(err)
}
return path.Join(home, ".ollama", "models", m.Name+".bin")
return filepath.Join(home, ".ollama", "models", m.Name+".bin")
}
func (m *Model) TempFile() string {
fullName := m.FullName()
return path.Join(
path.Dir(fullName),
fmt.Sprintf(".%s.part", path.Base(fullName)),
return filepath.Join(
filepath.Dir(fullName),
fmt.Sprintf(".%s.part", filepath.Base(fullName)),
)
}
@ -118,7 +118,7 @@ func saveModel(model *Model, fn func(total, completed int64)) error {
return os.Rename(model.TempFile(), model.FullName())
}
n , err := io.CopyN(out, resp.Body, 8192)
n, err := io.CopyN(out, resp.Body, 8192)
if err != nil && !errors.Is(err, io.EOF) {
return err
}

View file

@ -10,7 +10,7 @@ import (
"net"
"net/http"
"os"
"path"
"path/filepath"
"strings"
"text/template"
"time"
@ -32,7 +32,7 @@ func cacheDir() string {
panic(err)
}
return path.Join(home, ".ollama")
return filepath.Join(home, ".ollama")
}
func generate(c *gin.Context) {
@ -55,7 +55,7 @@ func generate(c *gin.Context) {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
req.Model = path.Join(cacheDir(), "models", req.Model+".bin")
req.Model = filepath.Join(cacheDir(), "models", req.Model+".bin")
}
templateNames := make([]string, 0, len(templates.Templates()))
@ -63,7 +63,7 @@ func generate(c *gin.Context) {
templateNames = append(templateNames, template.Name())
}
match, _ := matchRankOne(path.Base(req.Model), templateNames)
match, _ := matchRankOne(filepath.Base(req.Model), templateNames)
if template := templates.Lookup(match); template != nil {
var sb strings.Builder
if err := template.Execute(&sb, req); err != nil {