ollama/llm/shim_darwin.go

33 lines
727 B
Go
Raw Normal View History

package llm
import (
2023-12-19 02:32:04 +00:00
"embed"
"fmt"
2023-12-19 02:32:04 +00:00
"log"
"os"
"github.com/jmorganca/ollama/api"
)
2023-12-19 21:32:24 +00:00
//go:embed llama.cpp/gguf/ggml-metal.metal
2023-12-19 02:32:04 +00:00
var libEmbed embed.FS
func newRocmShimExtServer(model string, adapters, projectors []string, numLayers int64, opts api.Options) (extServer, error) {
// should never happen...
return nil, fmt.Errorf("ROCM GPUs not supported on Mac")
}
2023-12-19 02:32:04 +00:00
func nativeInit(workdir string) error {
2023-12-19 21:32:24 +00:00
err := extractLib(workdir, "llama.cpp/gguf/ggml-metal.metal")
2023-12-19 02:32:04 +00:00
if err != nil {
if err == payloadMissing {
// TODO perhaps consider this a hard failure on arm macs?
log.Printf("ggml-meta.metal payload missing")
return nil
}
return err
}
os.Setenv("GGML_METAL_PATH_RESOURCES", workdir)
return nil
}