From b9faf92148c7e6432ff0411783f444b7779fa36a Mon Sep 17 00:00:00 2001 From: Jonas Knobloch Date: Mon, 6 Apr 2026 22:59:50 +0200 Subject: Handle environment initialization on package level --- gpt2/onnx.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 gpt2/onnx.go (limited to 'gpt2/onnx.go') diff --git a/gpt2/onnx.go b/gpt2/onnx.go new file mode 100644 index 0000000..0da2ae2 --- /dev/null +++ b/gpt2/onnx.go @@ -0,0 +1,13 @@ +package gpt2 + +import ort "github.com/yalue/onnxruntime_go" + +func InitializeEnvironment() error { + ort.SetSharedLibraryPath(SharedLibraryPath()) + + return ort.InitializeEnvironment() +} + +func DestroyEnvironment() error { + return ort.DestroyEnvironment() +} -- cgit v1.3.1