summaryrefslogtreecommitdiff
path: root/gpt2/onnx.go
blob: 0da2ae224a36163151318753d45c7219644163ea (plain)
1
2
3
4
5
6
7
8
9
10
11
12
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()
}