diff --git a/registry/model.go b/registry/model.go index 6a84abed..f5f93ae3 100644 --- a/registry/model.go +++ b/registry/model.go @@ -1,5 +1,4 @@ package registry - import ( "bytes" "context" @@ -837,11 +836,19 @@ func InvokeGenerationFromSheet(wg *sync.WaitGroup, path string, modelsheetID, co } }() // Iterate models from the spreadsheet - for _, model := range modelCSVHelper.Models { - - if modelName != "" && modelName != model.Model { - continue - } + for _, model := range modelCSVHelper.Models + { + if modelName != "" && modelName != model.Model + { + continue + } + + if modelName == "" && strings.ToLower(strings.TrimSpace(model.PublishToRegistry)) != "true" + { + log.Infof("Skipping model: %s (PublishToRegistry != true)", model.Model) + continue + } + } totalAvailableModels++ ctx := context.Background()