From 1e02a5ec7f7794bb542e99ceb804ed88048b6b54 Mon Sep 17 00:00:00 2001 From: Jonas Knobloch Date: Fri, 24 Apr 2026 11:07:01 +0200 Subject: Access artifacts via shelf --- onnx/cmd/extract/main.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'onnx/cmd') diff --git a/onnx/cmd/extract/main.go b/onnx/cmd/extract/main.go index d8f8358..32b8c5d 100644 --- a/onnx/cmd/extract/main.go +++ b/onnx/cmd/extract/main.go @@ -5,11 +5,12 @@ import ( "log" "go.jknobloc.com/x/onnx" + "go.jknobloc.com/x/shelf" "go.jknobloc.com/x/tensor" ) func main() { - data, shape, err := onnx.ExtractInitializer("gpt2/models/base/model.onnx", "transformer.wte.weight") + data, shape, err := onnx.ExtractInitializer(shelf.Abs("models/gpt2/model.onnx"), "transformer.wte.weight") if err != nil { log.Fatal(err) -- cgit v1.3.1