vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-07-10 01:18:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

increase frame limit to 120 and allow loading tae ltx

Browse source
This commit is contained in:
Concedo 2026-05-29 18:50:01 +08:00
parent 8ddb4fedb1
commit 272c1ee232
3 changed files with 22 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
embd_res/kcpp_sdui.embd
View file

File diff suppressed because one or more lines are too long

2
koboldcpp.py
View file

@ -2806,7 +2806,7 @@ def sd_generate(genparams):
if flow_shift is not None and flow_shift < 0:
flow_shift = None # fall back to the default
sample_steps = (1 if sample_steps < 1 else (forced_steplimit if sample_steps > forced_steplimit else sample_steps))
vid_req_frames = (1 if vid_req_frames < 1 else (100 if vid_req_frames > 100 else vid_req_frames))
vid_req_frames = (1 if vid_req_frames < 1 else (120 if vid_req_frames > 120 else vid_req_frames))
swap_refimg = (True if tryparseint(genparams.get("send_as_refimg", 0),0) else False)
if len(extra_images_arr)==0 and swap_refimg and init_images and init_images!="" and not mask:

20
otherarch/sdcpp/stable-diffusion.cpp
View file

@ -285,6 +285,7 @@ public:
std::string t5_path_fixed = SAFE_STR(sd_ctx_params->t5xxl_path);
std::string taesd_path_fixed = SAFE_STR(sd_ctx_params->taesd_path);
std::string embed_connector_fixed = SAFE_STR(sd_ctx_params->embeddings_connectors_path);
std::string vae_path_fixed = SAFE_STR(sd_ctx_params->vae_path);
ModelLoader model_loader;
@ -483,6 +484,24 @@ public:
sd_ctx_params->photo_maker_path = "";
}
//for models with TAE suppport, if vae is set, tae is off, and it looks like a tae, swap to tae
if(taesd_path_fixed=="" && vae_path_fixed!="" && toLowerCase(vae_path_fixed).rfind("tae")!=std::string::npos)
{
try {
const uintmax_t tae_size_limit = 64 * 1024 * 1024; //if its less than 64mb, it might be a TAE
// Get the file size in bytes cross-platform
uintmax_t size = std::filesystem::file_size(vae_path_fixed);
if (size > 0 && size < tae_size_limit) {
printf("\nVAE appears to be a TAE, loading as TAE instead!\n");
taesd_path_fixed = vae_path_fixed;
vae_path_fixed = "";
}
}
catch (const std::filesystem::filesystem_error& e) {
std::printf("Error accessing file: %s\n", e.what());
}
}
sd_ctx_params->clip_g_path = clipg_path_fixed.c_str();
sd_ctx_params->clip_l_path = clipl_path_fixed.c_str();
sd_ctx_params->clip_vision_path = clip_vision_fixed.c_str();
@ -491,6 +510,7 @@ public:
sd_ctx_params->t5xxl_path = t5_path_fixed.c_str();
sd_ctx_params->taesd_path = taesd_path_fixed.c_str();
sd_ctx_params->embeddings_connectors_path = embed_connector_fixed.c_str();
sd_ctx_params->vae_path = vae_path_fixed.c_str();
//debug print
// printf("\n\nclip_g: %s\nclip_l: %s\nclip_vision: %s\nllm: %s\nllm_vision: %s\nt5xxl: %s\ntaesd: %s\n",
// sd_ctx_params->clip_g_path, sd_ctx_params->clip_l_path, sd_ctx_params->clip_vision_path,