From c51a83cfdb80933049543af41b795483a6032946 Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Tue, 14 Jul 2026 19:04:11 +0200
Subject: Refactor ONNX extraction

---
 research/sander/extract.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'research/sander')

diff --git a/research/sander/extract.go b/research/sander/extract.go
index c72fe7f..c9a2841 100644
--- a/research/sander/extract.go
+++ b/research/sander/extract.go
@@ -14,7 +14,15 @@ func (e *Experiment) Extract(db *sql.DB) error {
 	var data []float32
 	var shape []int
 
-	if d, s, err := onnx.ExtractInitializer(e.model, "transformer.wte.weight"); err != nil {
+	var model *onnx.Model
+
+	if m, err := onnx.NewModel(e.model); err != nil {
+		model = m
+	} else {
+		return err
+	}
+
+	if d, s, err := model.ExtractInitializer("transformer.wte.weight"); err != nil {
 		return err
 	} else {
 		data = d
-- 
cgit v1.3.1