mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2026-09-19 01:05:09 +02:00
sd: sync to master-660-d2797b8 (#2241)
* sd: sync to master-659-d3b2cb0 * sd: sync to master-660-d2797b8
This commit is contained in:
@@ -1225,14 +1225,20 @@ public:
|
||||
ggml_free(ctx);
|
||||
return false;
|
||||
}
|
||||
if (cond_stage_model) {
|
||||
cond_stage_model->alloc_params_buffer();
|
||||
if (cond_stage_model && !cond_stage_model->alloc_params_buffer()) {
|
||||
LOG_ERROR("Conditioner model params buffer allocation failed");
|
||||
ggml_free(ctx);
|
||||
return false;
|
||||
}
|
||||
if (diffusion_model) {
|
||||
diffusion_model->alloc_params_buffer();
|
||||
if (diffusion_model && !diffusion_model->alloc_params_buffer()) {
|
||||
LOG_ERROR("Diffusion model params buffer allocation failed");
|
||||
ggml_free(ctx);
|
||||
return false;
|
||||
}
|
||||
if (high_noise_diffusion_model) {
|
||||
high_noise_diffusion_model->alloc_params_buffer();
|
||||
if (high_noise_diffusion_model && !high_noise_diffusion_model->alloc_params_buffer()) {
|
||||
LOG_ERROR("High noise diffusion model params buffer allocation failed");
|
||||
ggml_free(ctx);
|
||||
return false;
|
||||
}
|
||||
if (first_stage_model && !first_stage_model->alloc_params_buffer()) {
|
||||
LOG_ERROR("VAE params buffer allocation failed");
|
||||
|
||||
Reference in New Issue
Block a user