Skip to content

Commit e84c27b

Browse files
committed
add set_txt2img_batch_count api
1 parent d0fda37 commit e84c27b

File tree

3 files changed

+56
-32
lines changed

3 files changed

+56
-32
lines changed

.idea/workspace.xml

Lines changed: 50 additions & 32 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

stable-diffusion-abi.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,10 @@ void set_txt2img_seed(sd_txt2img_options* opt, const int64_t seed) {
107107
opt->seed = seed;
108108
}
109109

110+
void set_txt2img_batch_count(sd_txt2img_options* opt, int batch_count) {
111+
opt->batch_count = batch_count;
112+
}
113+
110114
void set_img2img_init_img(sd_img2img_options* opt, const char* init_img) {
111115
const auto init_image = code::base64_decode<std::vector<uint8_t>, std::string>(init_img);
112116
opt->init_img = new uint8_t[init_image.size()];

stable-diffusion-abi.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,8 @@ STABLE_DIFFUSION_API void set_txt2img_sample_steps(sd_txt2img_options* opt, int
4545

4646
STABLE_DIFFUSION_API void set_txt2img_seed(sd_txt2img_options* opt, int64_t seed);
4747

48+
STABLE_DIFFUSION_API void set_txt2img_batch_count(sd_txt2img_options* opt, int batch_count);
49+
4850
//================================================================================
4951

5052
//==============================sd_img2img_options===============================

0 commit comments

Comments
 (0)