Skip to content

Commit

Permalink
chore: simplify model loading (#3715)
Browse files Browse the repository at this point in the history
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
  • Loading branch information
mudler authored Oct 2, 2024
1 parent 5488fc3 commit 3acd767
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions core/backend/tokenize.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,17 +11,11 @@ func ModelTokenize(s string, loader *model.ModelLoader, backendConfig config.Bac

modelFile := backendConfig.Model

grpcOpts := GRPCModelOpts(backendConfig)

var inferenceModel grpc.Backend
var err error

opts := modelOpts(backendConfig, appConfig, []model.Option{
model.WithLoadGRPCLoadModelOpts(grpcOpts),
model.WithThreads(uint32(*backendConfig.Threads)),
model.WithAssetDir(appConfig.AssetsDestination),
opts := ModelOptions(backendConfig, appConfig, []model.Option{
model.WithModel(modelFile),
model.WithContext(appConfig.Context),
})

if backendConfig.Backend == "" {
Expand Down

0 comments on commit 3acd767

Please sign in to comment.