Handle environment initialization on package level

1 files changed, 13 insertions, 0 deletions

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()
+}