mirror of
https://github.com/mudler/LocalAI.git
synced 2026-10-29 17:48:59 -04:00
mudler-agentandEttore Di Giacinto
0177e4988c
* feat(images): add upscale model capabilities * fix(config): keep SD GGML upscale opt-in * feat(stablediffusion): add lazy ESRGAN upscale bridge Keep native image ownership in C++ and serialize context creation, inference and teardown. Use upscale_scale in backend options as the explicit upscale-only marker because ModelOptions does not carry usecases. Assisted-by: Codex:gpt-5 * fix(stablediffusion): validate ESRGAN settings * fix(upscale): forward typed settings and contain resolved model paths Typed settings replace legacy options without duplicates. Resolve model and trusted root symlinks before caching the model path, rejecting missing files and escapes. Assisted-by: Codex:gpt-5 * fix(upscale): validate scale and clean failed HTTP outputs Require an explicit scale bounded to 1-16, check backend results, and keep destination cleanup owned by the allocating endpoint. Cover validation, result failures, cleanup, and successful URL responses. Assisted-by: Codex:gpt-5 * feat(studio): add inline image upscale workflow Assisted-by: OpenAI:gpt-5 * test(studio): serve upscale capability fixtures * fix(stablediffusion): default ESRGAN tiling to full frame * feat(gallery): add RealESRGAN anime upscaler * fix(stablediffusion): allow full-frame ESRGAN tiles * fix(routes): select explicit upscale models Give image upscaling dedicated usecase-filtered middleware and keep ordinary image models out of the endpoint. Mark and verify the legacy Diffusers x4 gallery model as an explicit upscaler. Assisted-by: OpenAI:gpt-5 [shell] * refactor(upscale): keep settings in backend options * test(config): keep upscale options backend-specific * fix: keep upscale routing metadata out of backend options --------- Co-authored-by: Ettore Di Giacinto <mudler@localai.io>
39 lines
2.2 KiB
C++
39 lines
2.2 KiB
C++
#include <cstdint>
|
|
#include "stable-diffusion.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
// Opaque ESRGAN context. Run returns 0, or argument/decode/inference/output/write
|
|
// errors (1..5); all image allocations remain native.
|
|
void* gosd_upscaler_create(const char* path, bool direct, int threads, int tile_size, const char* backend, const char* params_backend);
|
|
int gosd_upscaler_scale(void* handle);
|
|
int gosd_upscaler_run(void* handle, const char* src, const char* dst, int scale);
|
|
void gosd_upscaler_destroy(void* handle);
|
|
|
|
void sd_tiling_params_set_enabled(sd_tiling_params_t *params, bool enabled);
|
|
void sd_tiling_params_set_tile_sizes(sd_tiling_params_t *params, int tile_size_w, int tile_size_h);
|
|
void sd_tiling_params_set_rel_sizes(sd_tiling_params_t *params, float rel_size_w, float rel_size_h);
|
|
void sd_tiling_params_set_target_overlap(sd_tiling_params_t *params, float target_overlap);
|
|
sd_tiling_params_t* sd_img_gen_params_get_vae_tiling_params(sd_img_gen_params_t *params);
|
|
|
|
sd_img_gen_params_t* sd_img_gen_params_new(void);
|
|
void sd_img_gen_params_set_prompts(sd_img_gen_params_t *params, const char *prompt, const char *negative_prompt);
|
|
void sd_img_gen_params_set_dimensions(sd_img_gen_params_t *params, int width, int height);
|
|
void sd_img_gen_params_set_seed(sd_img_gen_params_t *params, int64_t seed);
|
|
void sd_img_gen_params_set_custom_sigmas(sd_img_gen_params_t *params, float *sigmas, int count);
|
|
|
|
int load_model(const char *model, char *model_path, char* options[], int threads, int diffusionModel);
|
|
int gen_image(sd_img_gen_params_t *p, int steps, char *dst, float cfg_scale, char *src_image, float strength, char *mask_image, char* ref_images[], int ref_images_count);
|
|
|
|
sd_vid_gen_params_t* sd_vid_gen_params_new(void);
|
|
void sd_vid_gen_params_set_prompts(sd_vid_gen_params_t *params, const char *prompt, const char *negative_prompt);
|
|
void sd_vid_gen_params_set_dimensions(sd_vid_gen_params_t *params, int width, int height);
|
|
void sd_vid_gen_params_set_seed(sd_vid_gen_params_t *params, int64_t seed);
|
|
void sd_vid_gen_params_set_video_frames(sd_vid_gen_params_t *params, int n);
|
|
int gen_video(sd_vid_gen_params_t *p, int steps, char *dst, float cfg_scale, int fps, char *init_image, char *end_image);
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|