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,