jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tensor loading iface

Browse Source
This commit is contained in:
Jesse Gross 2025-01-14 16:12:14 -08:00
parent 9722d96db2
commit a34960a516
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
model/model.go
View File

@ -152,6 +152,12 @@ func New(s string) (Model, error) {
} }
func populateFields(b ml.Backend, v reflect.Value, tags ...Tag) reflect.Value { func populateFields(b ml.Backend, v reflect.Value, tags ...Tag) reflect.Value {
var iface bool
if v.Kind() == reflect.Interface {
iface = true
v = v.Elem()
}
t := v.Type() t := v.Type()
if t.Kind() == reflect.Pointer { if t.Kind() == reflect.Pointer {
t, v = t.Elem(), v.Elem() t, v = t.Elem(), v.Elem()
@ -230,6 +236,10 @@ func populateFields(b ml.Backend, v reflect.Value, tags ...Tag) reflect.Value {
} }
} }
if iface {
return v.Addr()
}
return v return v
} }