Skip to content

Commit

Permalink
cmd: Create models dir if DNE
Browse files Browse the repository at this point in the history
  • Loading branch information
yondonfu committed Jan 25, 2024
1 parent 563d199 commit ac4729c
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions cmd/livepeer/starter/starter.go
Original file line number Diff line number Diff line change
Expand Up @@ -501,6 +501,11 @@ func StartLivepeer(ctx context.Context, cfg LivepeerConfig) {
}

modelDir := path.Join(*cfg.Datadir, "models")
if err := os.MkdirAll(modelDir, 0755); err != nil {
glog.Error("Error creating models dir %v", modelDir)
return
}

n.AIWorker, err = worker.NewWorker(aiWorkerContainerImageID, *cfg.Nvidia, modelDir)
if err != nil {
glog.Errorf("Error starting AI worker: %v", err)
Expand Down

0 comments on commit ac4729c

Please sign in to comment.