vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-05-06 16:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

allow wan to use img2img via init image

Browse source
This commit is contained in:
Concedo 2025-10-04 11:25:46 +08:00
parent db37688b47
commit bb06956b2d
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
otherarch/sdcpp/sdtype_adapter.cpp
View file

@ -682,9 +682,12 @@ sd_generation_outputs sdtype_generate(const sd_generation_inputs inputs)
sd_params->clip_skip = inputs.clip_skip;
sd_params->sample_method = sampler_from_name(inputs.sample_method);
bool is_img2img = img2img_data != "";
auto loadedsdver = get_loaded_sd_version(sd_ctx);
bool is_img2img = img2img_data != "";
bool is_wan = (loadedsdver == SDVersion::VERSION_WAN2 || loadedsdver == SDVersion::VERSION_WAN2_2_I2V || loadedsdver == SDVersion::VERSION_WAN2_2_TI2V);
bool is_qwenimg = (loadedsdver == SDVersion::VERSION_QWEN_IMAGE);
bool is_kontext = (loadedsdver==SDVersion::VERSION_FLUX && !loaded_model_is_chroma(sd_ctx));
if (loadedsdver == SDVersion::VERSION_FLUX)
{
if (!loaded_model_is_chroma(sd_ctx) && sd_params->cfg_scale != 1.0f) {
@ -703,6 +706,11 @@ sd_generation_outputs sdtype_generate(const sd_generation_inputs inputs)
}
}
if(is_wan && extra_image_data.size()==0 && is_img2img)
{
extra_image_data.push_back(img2img_data);
}
const int default_res_limit = 8192; // arbitrary, just to simplify the code
// avoid crashes due to bugs/limitations on certain models
// although it can be possible for a single side to exceed 1024, the total resolution of the image
@ -739,9 +747,6 @@ sd_generation_outputs sdtype_generate(const sd_generation_inputs inputs)
std::vector<sd_image_t> kontext_imgs;
std::vector<sd_image_t> wan_imgs;
std::vector<sd_image_t> photomaker_imgs;
bool is_wan = (loadedsdver == SDVersion::VERSION_WAN2 || loadedsdver == SDVersion::VERSION_WAN2_2_I2V || loadedsdver == SDVersion::VERSION_WAN2_2_TI2V);
bool is_qwenimg = (loadedsdver == SDVersion::VERSION_QWEN_IMAGE);
bool is_kontext = (loadedsdver==SDVersion::VERSION_FLUX && !loaded_model_is_chroma(sd_ctx));
if(is_qwenimg)
{