diff options
Diffstat (limited to 'gpt2')
| -rw-r--r-- | gpt2/cmd/gpt2/main.go | 3 | ||||
| -rw-r--r-- | gpt2/go.mod | 2 | ||||
| -rw-r--r-- | gpt2/model.go | 3 |
3 files changed, 5 insertions, 3 deletions
diff --git a/gpt2/cmd/gpt2/main.go b/gpt2/cmd/gpt2/main.go index b1e8611..69c7a21 100644 --- a/gpt2/cmd/gpt2/main.go +++ b/gpt2/cmd/gpt2/main.go @@ -2,8 +2,9 @@ package main import ( "fmt" - "gpt2" "log" + + "github.com/jonasknobloch/x/gpt2" ) func main() { diff --git a/gpt2/go.mod b/gpt2/go.mod index 2f58600..875375f 100644 --- a/gpt2/go.mod +++ b/gpt2/go.mod @@ -1,4 +1,4 @@ -module gpt2 +module github.com/jonasknobloch/x/gpt2 go 1.25.0 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" ) |