mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-09-23 21:46:20 +00:00
86 lines
2.8 KiB
C
86 lines
2.8 KiB
C
#ifndef CLIP_H
|
|
#define CLIP_H
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
#ifdef LLAMA_SHARED
|
|
# if defined(_WIN32) && !defined(__MINGW32__)
|
|
# ifdef LLAMA_BUILD
|
|
# define CLIP_API __declspec(dllexport)
|
|
# else
|
|
# define CLIP_API __declspec(dllimport)
|
|
# endif
|
|
# else
|
|
# define CLIP_API __attribute__ ((visibility ("default")))
|
|
# endif
|
|
#else
|
|
# define CLIP_API
|
|
#endif
|
|
|
|
struct clip_ctx;
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
struct clip_ctx;
|
|
|
|
struct clip_image_u8_batch {
|
|
struct clip_image_u8 * data;
|
|
size_t size;
|
|
};
|
|
|
|
struct clip_image_f32_batch {
|
|
struct clip_image_f32 * data;
|
|
size_t size;
|
|
};
|
|
|
|
CLIP_API struct clip_ctx * clip_model_load (const char * fname, int verbosity);
|
|
CLIP_API struct clip_ctx * clip_model_load_cpu(const char * fname, int verbosity);
|
|
|
|
CLIP_API void clip_free(struct clip_ctx * ctx);
|
|
|
|
CLIP_API size_t clip_embd_nbytes(const struct clip_ctx * ctx);
|
|
|
|
CLIP_API int32_t clip_image_size (const struct clip_ctx * ctx);
|
|
CLIP_API int32_t clip_patch_size (const struct clip_ctx * ctx);
|
|
CLIP_API int32_t clip_hidden_size(const struct clip_ctx * ctx);
|
|
|
|
// TODO: should be enum, not string
|
|
CLIP_API const char * clip_patch_merge_type(const struct clip_ctx * ctx);
|
|
|
|
CLIP_API const int32_t * clip_image_grid(const struct clip_ctx * ctx);
|
|
|
|
CLIP_API int clip_n_patches (const struct clip_ctx * ctx);
|
|
CLIP_API int clip_n_mmproj_embd(const struct clip_ctx * ctx);
|
|
|
|
CLIP_API struct clip_image_u8 * clip_image_u8_init ();
|
|
CLIP_API struct clip_image_f32 * clip_image_f32_init();
|
|
|
|
CLIP_API void clip_image_u8_free (struct clip_image_u8 * img);
|
|
CLIP_API void clip_image_f32_free(struct clip_image_f32 * img);
|
|
CLIP_API void clip_image_u8_batch_free (struct clip_image_u8_batch * batch);
|
|
CLIP_API void clip_image_f32_batch_free(struct clip_image_f32_batch * batch);
|
|
|
|
CLIP_API bool clip_image_load_from_file(const char * fname, struct clip_image_u8 * img);
|
|
|
|
/** interpret bytes as an image file with length bytes_length, and use the result to populate img */
|
|
CLIP_API bool clip_image_load_from_bytes(const unsigned char * bytes, size_t bytes_length, struct clip_image_u8 * img);
|
|
|
|
/** preprocess img and store the result in res_imgs, pad_to_square may be overriden to false depending on model configuration */
|
|
CLIP_API bool clip_image_preprocess(struct clip_ctx * ctx, const struct clip_image_u8 * img, struct clip_image_f32_batch * res_imgs );
|
|
|
|
CLIP_API struct ggml_tensor * clip_get_newline_tensor(const struct clip_ctx * ctx);
|
|
|
|
CLIP_API bool clip_image_encode (struct clip_ctx * ctx, int n_threads, struct clip_image_f32 * img, float * vec);
|
|
CLIP_API bool clip_image_batch_encode(struct clip_ctx * ctx, int n_threads, const struct clip_image_f32_batch * imgs, float * vec);
|
|
|
|
CLIP_API bool clip_model_quantize(const char * fname_inp, const char * fname_out, int itype);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // CLIP_H
|