From a34960a5167a00f7ed2a04f4e3944ae5ca6faa6e Mon Sep 17 00:00:00 2001 From: Jesse Gross Date: Tue, 14 Jan 2025 16:12:14 -0800 Subject: [PATCH] tensor loading iface --- model/model.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/model/model.go b/model/model.go index 8f177f0eb..507910ce1 100644 --- a/model/model.go +++ b/model/model.go @@ -152,6 +152,12 @@ func New(s string) (Model, error) { } 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() if t.Kind() == reflect.Pointer { 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 }