cmd: Create models dir if DNE

livepeer · Jan 25, 2024 · ac4729c · ac4729c
1 parent 563d199
commit ac4729c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/livepeer/starter/starter.go b/cmd/livepeer/starter/starter.go
@@ -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)