summaryrefslogtreecommitdiff
path: root/gpt2/model.go
diff options
context:
space:
mode:
Diffstat (limited to 'gpt2/model.go')
-rw-r--r--gpt2/model.go3
1 files changed, 2 insertions, 1 deletions
diff --git a/gpt2/model.go b/gpt2/model.go
index 79a4407..48b052c 100644
--- a/gpt2/model.go
+++ b/gpt2/model.go
@@ -3,13 +3,14 @@ package gpt2
import (
"errors"
"fmt"
- _ "llm"
"math"
"os"
"slices"
"sort"
"strconv"
+ _ "github.com/jonasknobloch/x/llm"
+
ort "github.com/yalue/onnxruntime_go"
)