mirror of
https://github.com/ollama/ollama.git
synced 2026-04-23 01:05:47 +02:00
56c735d871
Remove the vendored GGML and llama.cpp backend, CGO runner, Go model implementations, and sample. llama-server (built from upstream llama.cpp via FetchContent) is now the sole inference engine for GGUF-based models. (Safetensor based models continue to run on the new MLX engine.) This allows us to more rapidly pick up new capabilities and fixes from llama.cpp as they come out. On windows this now requires recent AMD driver versions to support ROCm v7 as llama.cpp currently does not support building against v6.
104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
package convert
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"io/fs"
|
|
"strings"
|
|
)
|
|
|
|
type Tensor interface {
|
|
Name() string
|
|
Shape() []uint64
|
|
Kind() uint32
|
|
SetRepacker(Repacker)
|
|
WriteTo(io.Writer) (int64, error)
|
|
Clone() Tensor
|
|
}
|
|
|
|
type tensorBase struct {
|
|
name string
|
|
shape []uint64
|
|
repacker Repacker
|
|
}
|
|
|
|
func (t tensorBase) Name() string {
|
|
return t.name
|
|
}
|
|
|
|
func (t tensorBase) Shape() []uint64 {
|
|
return t.shape
|
|
}
|
|
|
|
const (
|
|
tensorKindFP32 uint32 = iota
|
|
tensorKindFP16
|
|
tensorKindBF16 = 30
|
|
tensorKindMXFP4 = 39
|
|
)
|
|
|
|
func (t tensorBase) Kind() uint32 {
|
|
if strings.HasSuffix(t.name, ".ffn_gate_inp.weight") ||
|
|
strings.HasSuffix(t.name, ".bias") ||
|
|
strings.HasSuffix(t.name, ".shortconv.conv.weight") ||
|
|
strings.HasSuffix(t.name, ".ssm_conv1d.weight") || // SSM conv kernel must be F32 for Metal
|
|
strings.HasPrefix(t.name, "a.conv1d.") || // audio SSCP conv weights must be F32 for im2col
|
|
strings.Contains(t.name, ".conv_dw.") || // audio depthwise conv weights must be F32
|
|
t.name == "token_types.weight" ||
|
|
t.name == "v.positional_embedding_vlm" ||
|
|
t.name == "v.patch_embd.weight" ||
|
|
t.name == "v.patch_embedding.weight" ||
|
|
t.name == "v.patch_conv.weight" ||
|
|
t.name == "v.position_embd.weight" ||
|
|
t.name == "v.position_embedding.weight" ||
|
|
t.name == "v.tile_position_embd.weight" ||
|
|
t.name == "v.pre_tile_position_embd.weight" ||
|
|
t.name == "v.post_tile_position_embd.weight" ||
|
|
t.name == "s.position_embd" ||
|
|
strings.HasSuffix(t.name, "rel_pos_h") ||
|
|
strings.HasSuffix(t.name, "rel_pos_w") {
|
|
// these tensors are always F32
|
|
return tensorKindFP32
|
|
}
|
|
|
|
switch len(t.shape) {
|
|
case 0:
|
|
panic("invalid tensor shape")
|
|
case 1:
|
|
return tensorKindFP32
|
|
default:
|
|
return tensorKindFP16
|
|
}
|
|
}
|
|
|
|
func (t *tensorBase) SetRepacker(fn Repacker) {
|
|
t.repacker = fn
|
|
}
|
|
|
|
type Repacker func(string, []float32, []uint64) ([]float32, error)
|
|
|
|
func parseTensors(fsys fs.FS, replacer *strings.Replacer) ([]Tensor, error) {
|
|
patterns := []struct {
|
|
Pattern string
|
|
Func func(fs.FS, *strings.Replacer, ...string) ([]Tensor, error)
|
|
}{
|
|
{"*.safetensors", parseSafetensors},
|
|
{"pytorch_model-*-of-*.bin", parseTorch},
|
|
{"pytorch_model.bin", parseTorch},
|
|
{"consolidated.*.pth", parseTorch},
|
|
}
|
|
|
|
for _, pattern := range patterns {
|
|
matches, err := fs.Glob(fsys, pattern.Pattern)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(matches) > 0 {
|
|
return pattern.Func(fsys, replacer, matches...)
|
|
}
|
|
}
|
|
|
|
return nil, errors.New("unknown tensor format")
|
|
}
|