diff --git a/pkg/converter/converter.go b/pkg/converter/converter.go index f787b6c..2976d67 100644 --- a/pkg/converter/converter.go +++ b/pkg/converter/converter.go @@ -6,6 +6,7 @@ package converter import ( "fmt" + "github.com/nlpodyssey/cybertron/pkg/converter/bart" "github.com/nlpodyssey/cybertron/pkg/converter/bert" "github.com/nlpodyssey/cybertron/pkg/models" diff --git a/pkg/downloader/downloadmodel.go b/pkg/downloader/downloadmodel.go index f69e8e5..b37ef56 100644 --- a/pkg/downloader/downloadmodel.go +++ b/pkg/downloader/downloadmodel.go @@ -6,12 +6,13 @@ package downloader import ( "fmt" - "github.com/nlpodyssey/cybertron/pkg/models" - "github.com/rs/zerolog/log" "io" "net/http" "os" "path/filepath" + + "github.com/nlpodyssey/cybertron/pkg/models" + "github.com/rs/zerolog/log" ) const (