summaryrefslogtreecommitdiff
path: root/gpt2/model.go
diff options
context:
space:
mode:
Diffstat (limited to 'gpt2/model.go')
-rw-r--r--gpt2/model.go10
1 files changed, 1 insertions, 9 deletions
diff --git a/gpt2/model.go b/gpt2/model.go
index f492c56..d0d1c2f 100644
--- a/gpt2/model.go
+++ b/gpt2/model.go
@@ -60,12 +60,6 @@ func IntraOpNumThreads() int {
}
func (m *Model) Init() error {
- ort.SetSharedLibraryPath(SharedLibraryPath())
-
- if err := ort.InitializeEnvironment(); err != nil {
- return err
- }
-
m.allocator = NewAllocator(m.config, m.withCache, m.withLogits, m.withLogProbs)
var options *ort.SessionOptions
@@ -99,10 +93,8 @@ func (m *Model) Init() error {
return nil
}
-func (m *Model) Destroy() error {
+func (m *Model) Destroy() {
m.allocator.Destroy()
-
- return ort.DestroyEnvironment()
}
func (m *Model) Generate(prompt []int64, steps int64, logits *[][]float32) ([]int64, error) {