diff --git a/src/GenerateTab.vue b/src/GenerateTab.vue index e38e629..3b66172 100644 --- a/src/GenerateTab.vue +++ b/src/GenerateTab.vue @@ -163,6 +163,13 @@ + + + Number of image batches + {{ imagesNumberBatch }} + + +
Resize @@ -285,6 +292,7 @@ export default { minimumDimension: 512, maximumDimension: 512, imagesNumber: 4, + imagesNumberBatch: 1, styles: [], generatedImages: [], @@ -397,6 +405,7 @@ export default { this.cfgScale = storage.localStorage.getItem('cfgScale') || this.cfgScale; this.currentSampler = storage.localStorage.getItem('currentSampler') || this.currentSampler; this.imagesNumber = storage.localStorage.getItem('imagesNumber') || this.imagesNumber; + this.imagesNumberBatch = storage.localStorage.getItem('imagesNumberBatch') || this.imagesNumberBatch; this.currentMode = storage.localStorage.getItem('currentMode') || this.currentMode; this.isSaveImagesLocally = storage.localStorage.getItem('isSaveImagesLocally') || this.isSaveImagesLocally; diff --git a/src/maskGeneratorMixin.js b/src/maskGeneratorMixin.js index 3112eea..5703d34 100644 --- a/src/maskGeneratorMixin.js +++ b/src/maskGeneratorMixin.js @@ -164,7 +164,7 @@ export default { seed_resize_from_h: -1, seed_resize_from_w: -1, batch_size: this.imagesNumber, - n_iter: 1, + n_iter: this.imagesNumberBatch, steps: Number(this.steps), cfg_scale: Number(this.cfgScale), width: this.getSizeForGeneratingImage.width, @@ -199,7 +199,7 @@ export default { seed_resize_from_h: -1, seed_resize_from_w: -1, batch_size: this.imagesNumber, - n_iter: 1, + n_iter: this.imagesNumberBatch, steps: Number(this.steps), cfg_scale: Number(this.cfgScale), width: this.getSizeForGeneratingImage.width,