diff options
| author | Jonas Knobloch <jonas.knobloch@t-online.de> | 2026-04-24 11:07:01 +0200 |
|---|---|---|
| committer | Jonas Knobloch <jonas.knobloch@t-online.de> | 2026-04-24 11:07:01 +0200 |
| commit | 1e02a5ec7f7794bb542e99ceb804ed88048b6b54 (patch) | |
| tree | 5db849f4eaa326f9b6511c315180e540f00c393e /onnx/cmd/extract | |
| parent | 875ac19922932b12f6357d64a460c06d031000e7 (diff) | |
Access artifacts via shelf
Diffstat (limited to 'onnx/cmd/extract')
| -rw-r--r-- | onnx/cmd/extract/main.go | 3 |
1 files changed, 2 insertions, 1 deletions
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) |