@fugood/llama.node 0.3.15 → 0.3.17
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CMakeLists.txt +3 -0
- package/bin/darwin/arm64/llama-node.node +0 -0
- package/bin/darwin/x64/llama-node.node +0 -0
- package/bin/linux/arm64/llama-node.node +0 -0
- package/bin/linux/x64/llama-node.node +0 -0
- package/bin/linux-cuda/arm64/llama-node.node +0 -0
- package/bin/linux-cuda/x64/llama-node.node +0 -0
- package/bin/linux-vulkan/arm64/llama-node.node +0 -0
- package/bin/linux-vulkan/x64/llama-node.node +0 -0
- package/bin/win32/arm64/llama-node.node +0 -0
- package/bin/win32/arm64/node.lib +0 -0
- package/bin/win32/x64/llama-node.node +0 -0
- package/bin/win32/x64/node.lib +0 -0
- package/bin/win32-vulkan/arm64/llama-node.node +0 -0
- package/bin/win32-vulkan/arm64/node.lib +0 -0
- package/bin/win32-vulkan/x64/llama-node.node +0 -0
- package/bin/win32-vulkan/x64/node.lib +0 -0
- package/lib/binding.ts +5 -0
- package/package.json +1 -1
- package/src/LlamaCompletionWorker.cpp +8 -0
- package/src/LlamaCompletionWorker.h +1 -0
- package/src/LlamaContext.cpp +3 -2
- package/src/llama.cpp/.github/workflows/build-linux-cross.yml +124 -0
- package/src/llama.cpp/.github/workflows/build.yml +70 -27
- package/src/llama.cpp/.github/workflows/docker.yml +6 -6
- package/src/llama.cpp/.github/workflows/server.yml +7 -11
- package/src/llama.cpp/CMakeLists.txt +23 -1
- package/src/llama.cpp/common/CMakeLists.txt +6 -3
- package/src/llama.cpp/common/arg.cpp +809 -105
- package/src/llama.cpp/common/arg.h +9 -0
- package/src/llama.cpp/common/chat.cpp +1 -1
- package/src/llama.cpp/common/common.cpp +31 -521
- package/src/llama.cpp/common/common.h +17 -36
- package/src/llama.cpp/common/json-schema-to-grammar.cpp +3 -0
- package/src/llama.cpp/common/llguidance.cpp +30 -47
- package/src/llama.cpp/common/minja/chat-template.hpp +15 -7
- package/src/llama.cpp/common/minja/minja.hpp +119 -93
- package/src/llama.cpp/common/sampling.cpp +3 -0
- package/src/llama.cpp/docs/build.md +122 -7
- package/src/llama.cpp/examples/CMakeLists.txt +0 -9
- package/src/llama.cpp/examples/batched/batched.cpp +1 -1
- package/src/llama.cpp/examples/batched-bench/batched-bench.cpp +1 -1
- package/src/llama.cpp/examples/embedding/embedding.cpp +7 -1
- package/src/llama.cpp/examples/export-lora/export-lora.cpp +1 -1
- package/src/llama.cpp/examples/gguf-split/gguf-split.cpp +15 -16
- package/src/llama.cpp/examples/gritlm/gritlm.cpp +1 -1
- package/src/llama.cpp/examples/llama-bench/llama-bench.cpp +210 -8
- package/src/llama.cpp/examples/llama.android/llama/build.gradle.kts +1 -0
- package/src/llama.cpp/examples/llava/CMakeLists.txt +39 -24
- package/src/llama.cpp/examples/llava/clip-impl.h +345 -0
- package/src/llama.cpp/examples/llava/clip.cpp +2152 -1803
- package/src/llama.cpp/examples/llava/clip.h +39 -22
- package/src/llama.cpp/examples/llava/deprecation-warning.cpp +22 -0
- package/src/llama.cpp/examples/llava/llava.cpp +64 -52
- package/src/llama.cpp/examples/llava/mtmd-cli.cpp +344 -0
- package/src/llama.cpp/examples/llava/mtmd.cpp +708 -0
- package/src/llama.cpp/examples/llava/mtmd.h +168 -0
- package/src/llama.cpp/examples/llava/{qwen2vl-cli.cpp → qwen2vl-test.cpp} +83 -31
- package/src/llama.cpp/examples/main/main.cpp +16 -5
- package/src/llama.cpp/examples/parallel/parallel.cpp +3 -1
- package/src/llama.cpp/examples/passkey/passkey.cpp +1 -1
- package/src/llama.cpp/examples/perplexity/perplexity.cpp +17 -3
- package/src/llama.cpp/examples/quantize/quantize.cpp +115 -2
- package/src/llama.cpp/examples/rpc/CMakeLists.txt +4 -2
- package/src/llama.cpp/examples/rpc/rpc-server.cpp +163 -8
- package/src/llama.cpp/examples/run/CMakeLists.txt +12 -1
- package/src/llama.cpp/examples/run/run.cpp +14 -28
- package/src/llama.cpp/examples/server/httplib.h +313 -247
- package/src/llama.cpp/examples/server/server.cpp +243 -139
- package/src/llama.cpp/examples/server/utils.hpp +51 -2
- package/src/llama.cpp/examples/speculative/speculative.cpp +1 -1
- package/src/llama.cpp/examples/speculative-simple/speculative-simple.cpp +1 -1
- package/src/llama.cpp/examples/sycl/build.sh +2 -2
- package/src/llama.cpp/examples/sycl/win-build-sycl.bat +2 -2
- package/src/llama.cpp/examples/tts/tts.cpp +14 -9
- package/src/llama.cpp/ggml/CMakeLists.txt +8 -2
- package/src/llama.cpp/ggml/cmake/GitVars.cmake +22 -0
- package/src/llama.cpp/ggml/include/ggml-cpu.h +5 -0
- package/src/llama.cpp/ggml/include/ggml-rpc.h +6 -1
- package/src/llama.cpp/ggml/include/ggml.h +66 -99
- package/src/llama.cpp/ggml/src/CMakeLists.txt +15 -8
- package/src/llama.cpp/ggml/src/ggml-cann/CMakeLists.txt +0 -2
- package/src/llama.cpp/ggml/src/ggml-cann/acl_tensor.cpp +8 -4
- package/src/llama.cpp/ggml/src/ggml-cann/acl_tensor.h +5 -5
- package/src/llama.cpp/ggml/src/ggml-cann/aclnn_ops.cpp +692 -1534
- package/src/llama.cpp/ggml/src/ggml-cann/aclnn_ops.h +613 -122
- package/src/llama.cpp/ggml/src/ggml-cann/common.h +135 -1
- package/src/llama.cpp/ggml/src/ggml-cann/ggml-cann.cpp +507 -137
- package/src/llama.cpp/ggml/src/ggml-common.h +12 -6
- package/src/llama.cpp/ggml/src/ggml-cpu/CMakeLists.txt +48 -22
- package/src/llama.cpp/ggml/src/ggml-cpu/binary-ops.cpp +158 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/binary-ops.h +16 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/common.h +72 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/cpu-feats-x86.cpp +1 -1
- package/src/llama.cpp/ggml/src/ggml-cpu/ggml-cpu-aarch64.cpp +2413 -228
- package/src/llama.cpp/ggml/src/ggml-cpu/ggml-cpu-impl.h +2 -21
- package/src/llama.cpp/ggml/src/ggml-cpu/ggml-cpu-quants.c +754 -404
- package/src/llama.cpp/ggml/src/ggml-cpu/ggml-cpu.c +1004 -13516
- package/src/llama.cpp/ggml/src/ggml-cpu/ggml-cpu.cpp +2 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/kleidiai/kernels.cpp +2 -7
- package/src/llama.cpp/ggml/src/ggml-cpu/kleidiai/kernels.h +0 -1
- package/src/llama.cpp/ggml/src/ggml-cpu/kleidiai/kleidiai.cpp +3 -4
- package/src/llama.cpp/ggml/src/ggml-cpu/llamafile/sgemm.cpp +533 -88
- package/src/llama.cpp/ggml/src/ggml-cpu/ops.cpp +8809 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/ops.h +110 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/simd-mappings.h +892 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/unary-ops.cpp +186 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/unary-ops.h +28 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/vec.cpp +258 -0
- package/src/llama.cpp/ggml/src/ggml-cpu/vec.h +802 -0
- package/src/llama.cpp/ggml/src/ggml-cuda/vendors/hip.h +7 -0
- package/src/llama.cpp/ggml/src/ggml-cuda/vendors/musa.h +1 -0
- package/src/llama.cpp/ggml/src/ggml-hip/CMakeLists.txt +0 -4
- package/src/llama.cpp/ggml/src/ggml-impl.h +52 -18
- package/src/llama.cpp/ggml/src/ggml-metal/ggml-metal-impl.h +70 -3
- package/src/llama.cpp/ggml/src/ggml-opencl/CMakeLists.txt +67 -119
- package/src/llama.cpp/ggml/src/ggml-opencl/ggml-opencl.cpp +1023 -260
- package/src/llama.cpp/ggml/src/ggml-rpc/ggml-rpc.cpp +293 -40
- package/src/llama.cpp/ggml/src/ggml-sycl/CMakeLists.txt +127 -33
- package/src/llama.cpp/ggml/src/ggml-sycl/backend.hpp +1 -0
- package/src/llama.cpp/ggml/src/ggml-sycl/binbcast.cpp +350 -0
- package/src/llama.cpp/ggml/src/ggml-sycl/binbcast.hpp +39 -0
- package/src/llama.cpp/ggml/src/ggml-sycl/common.cpp +0 -35
- package/src/llama.cpp/ggml/src/ggml-sycl/common.hpp +29 -293
- package/src/llama.cpp/ggml/src/ggml-sycl/dpct/helper.hpp +79 -90
- package/src/llama.cpp/ggml/src/ggml-sycl/element_wise.cpp +967 -438
- package/src/llama.cpp/ggml/src/ggml-sycl/element_wise.hpp +22 -23
- package/src/llama.cpp/ggml/src/ggml-sycl/gemm.hpp +12 -43
- package/src/llama.cpp/ggml/src/ggml-sycl/getrows.cpp +24 -20
- package/src/llama.cpp/ggml/src/ggml-sycl/getrows.hpp +1 -4
- package/src/llama.cpp/ggml/src/ggml-sycl/ggml-sycl.cpp +210 -286
- package/src/llama.cpp/ggml/src/ggml-sycl/im2col.cpp +84 -74
- package/src/llama.cpp/ggml/src/ggml-sycl/im2col.hpp +1 -3
- package/src/llama.cpp/ggml/src/ggml-sycl/norm.cpp +37 -49
- package/src/llama.cpp/ggml/src/ggml-sycl/norm.hpp +7 -22
- package/src/llama.cpp/ggml/src/ggml-sycl/outprod.cpp +4 -14
- package/src/llama.cpp/ggml/src/ggml-sycl/rope.cpp +204 -118
- package/src/llama.cpp/ggml/src/ggml-sycl/rope.hpp +1 -3
- package/src/llama.cpp/ggml/src/ggml-vulkan/CMakeLists.txt +23 -0
- package/src/llama.cpp/ggml/src/ggml-vulkan/ggml-vulkan.cpp +692 -126
- package/src/llama.cpp/ggml/src/ggml-vulkan/vulkan-shaders/CMakeLists.txt +12 -0
- package/src/llama.cpp/ggml/src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp +21 -10
- package/src/llama.cpp/ggml/src/ggml.c +141 -245
- package/src/llama.cpp/ggml/src/gguf.cpp +1 -0
- package/src/llama.cpp/include/llama.h +30 -11
- package/src/llama.cpp/models/ggml-vocab-llama4.gguf.inp +112 -0
- package/src/llama.cpp/models/ggml-vocab-llama4.gguf.out +46 -0
- package/src/llama.cpp/models/ggml-vocab-pixtral.gguf.inp +112 -0
- package/src/llama.cpp/models/ggml-vocab-pixtral.gguf.out +46 -0
- package/src/llama.cpp/requirements/requirements-all.txt +2 -0
- package/src/llama.cpp/requirements/requirements-gguf_editor_gui.txt +3 -0
- package/src/llama.cpp/src/CMakeLists.txt +3 -2
- package/src/llama.cpp/src/llama-adapter.cpp +37 -1
- package/src/llama.cpp/src/llama-arch.cpp +161 -17
- package/src/llama.cpp/src/llama-arch.h +16 -0
- package/src/llama.cpp/src/llama-chat.cpp +82 -17
- package/src/llama.cpp/src/llama-chat.h +6 -2
- package/src/llama.cpp/src/llama-context.cpp +108 -92
- package/src/llama.cpp/src/llama-context.h +1 -2
- package/src/llama.cpp/src/llama-graph.cpp +189 -119
- package/src/llama.cpp/src/llama-graph.h +26 -6
- package/src/llama.cpp/src/llama-hparams.h +13 -0
- package/src/llama.cpp/src/llama-kv-cache.cpp +70 -123
- package/src/llama.cpp/src/llama-kv-cache.h +41 -115
- package/src/llama.cpp/src/llama-memory.h +1 -1
- package/src/llama.cpp/src/llama-mmap.cpp +1 -1
- package/src/llama.cpp/src/llama-model-loader.cpp +10 -5
- package/src/llama.cpp/src/llama-model-loader.h +5 -3
- package/src/llama.cpp/src/llama-model.cpp +1544 -291
- package/src/llama.cpp/src/llama-model.h +13 -1
- package/src/llama.cpp/src/llama-quant.cpp +29 -8
- package/src/llama.cpp/src/llama-sampling.cpp +7 -1
- package/src/llama.cpp/src/llama-vocab.cpp +44 -6
- package/src/llama.cpp/src/llama.cpp +1 -1
- package/src/llama.cpp/tests/CMakeLists.txt +43 -30
- package/src/llama.cpp/tests/test-arg-parser.cpp +51 -4
- package/src/llama.cpp/tests/test-backend-ops.cpp +139 -57
- package/src/llama.cpp/tests/test-chat-template.cpp +34 -13
- package/src/llama.cpp/tests/test-chat.cpp +12 -2
- package/src/llama.cpp/{examples/gbnf-validator/gbnf-validator.cpp → tests/test-gbnf-validator.cpp} +2 -2
- package/src/llama.cpp/tests/test-grammar-integration.cpp +3 -2
- package/src/llama.cpp/tests/test-grammar-llguidance.cpp +63 -2
- package/src/llama.cpp/tests/test-grammar-parser.cpp +3 -1
- package/src/llama.cpp/tests/test-json-schema-to-grammar.cpp +17 -1
- package/src/llama.cpp/tests/test-llama-grammar.cpp +2 -1
- package/src/llama.cpp/{examples/quantize-stats/quantize-stats.cpp → tests/test-quantize-stats.cpp} +3 -1
- package/src/llama.cpp/tests/test-tokenizer-1-bpe.cpp +2 -1
- package/src/llama.cpp/tests/test-tokenizer-1-spm.cpp +2 -1
- package/src/llama.cpp/examples/gbnf-validator/CMakeLists.txt +0 -5
- package/src/llama.cpp/examples/llava/gemma3-cli.cpp +0 -341
- package/src/llama.cpp/examples/llava/llava-cli.cpp +0 -332
- package/src/llama.cpp/examples/llava/minicpmv-cli.cpp +0 -354
- package/src/llama.cpp/examples/quantize-stats/CMakeLists.txt +0 -6
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/CMakeLists.txt +0 -30
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/ascendc_kernels.h +0 -19
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/dup.cpp +0 -234
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/get_row_f16.cpp +0 -197
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/get_row_f32.cpp +0 -190
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/get_row_q4_0.cpp +0 -204
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/get_row_q8_0.cpp +0 -191
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/quantize_f16_q8_0.cpp +0 -218
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/quantize_f32_q8_0.cpp +0 -216
- package/src/llama.cpp/ggml/src/ggml-cann/kernels/quantize_float_to_q4_0.cpp +0 -295
|
@@ -1,332 +0,0 @@
|
|
|
1
|
-
#include "arg.h"
|
|
2
|
-
#include "base64.hpp"
|
|
3
|
-
#include "log.h"
|
|
4
|
-
#include "common.h"
|
|
5
|
-
#include "sampling.h"
|
|
6
|
-
#include "clip.h"
|
|
7
|
-
#include "llava.h"
|
|
8
|
-
#include "llama.h"
|
|
9
|
-
#include "ggml.h"
|
|
10
|
-
|
|
11
|
-
#include <cstdio>
|
|
12
|
-
#include <cstdlib>
|
|
13
|
-
#include <cstring>
|
|
14
|
-
#include <vector>
|
|
15
|
-
|
|
16
|
-
static bool eval_tokens(struct llama_context * ctx_llama, std::vector<llama_token> tokens, int n_batch, int * n_past) {
|
|
17
|
-
int N = (int) tokens.size();
|
|
18
|
-
for (int i = 0; i < N; i += n_batch) {
|
|
19
|
-
int n_eval = (int) tokens.size() - i;
|
|
20
|
-
if (n_eval > n_batch) {
|
|
21
|
-
n_eval = n_batch;
|
|
22
|
-
}
|
|
23
|
-
if (llama_decode(ctx_llama, llama_batch_get_one(&tokens[i], n_eval))) {
|
|
24
|
-
LOG_ERR("%s : failed to eval. token %d/%d (batch size %d, n_past %d)\n", __func__, i, N, n_batch, *n_past);
|
|
25
|
-
return false;
|
|
26
|
-
}
|
|
27
|
-
*n_past += n_eval;
|
|
28
|
-
}
|
|
29
|
-
return true;
|
|
30
|
-
}
|
|
31
|
-
|
|
32
|
-
static bool eval_id(struct llama_context * ctx_llama, int id, int * n_past) {
|
|
33
|
-
std::vector<llama_token> tokens;
|
|
34
|
-
tokens.push_back(id);
|
|
35
|
-
return eval_tokens(ctx_llama, tokens, 1, n_past);
|
|
36
|
-
}
|
|
37
|
-
|
|
38
|
-
static bool eval_string(struct llama_context * ctx_llama, const char* str, int n_batch, int * n_past, bool add_bos){
|
|
39
|
-
std::string str2 = str;
|
|
40
|
-
std::vector<llama_token> embd_inp = common_tokenize(ctx_llama, str2, add_bos, true);
|
|
41
|
-
eval_tokens(ctx_llama, embd_inp, n_batch, n_past);
|
|
42
|
-
return true;
|
|
43
|
-
}
|
|
44
|
-
|
|
45
|
-
static const char * sample(struct common_sampler * smpl,
|
|
46
|
-
struct llama_context * ctx_llama,
|
|
47
|
-
int * n_past) {
|
|
48
|
-
const llama_token id = common_sampler_sample(smpl, ctx_llama, -1);
|
|
49
|
-
common_sampler_accept(smpl, id, true);
|
|
50
|
-
|
|
51
|
-
const llama_model * model = llama_get_model(ctx_llama);
|
|
52
|
-
const llama_vocab * vocab = llama_model_get_vocab(model);
|
|
53
|
-
|
|
54
|
-
static std::string ret;
|
|
55
|
-
if (llama_vocab_is_eog(vocab, id)) {
|
|
56
|
-
ret = "</s>";
|
|
57
|
-
} else {
|
|
58
|
-
ret = common_token_to_piece(ctx_llama, id);
|
|
59
|
-
}
|
|
60
|
-
eval_id(ctx_llama, id, n_past);
|
|
61
|
-
return ret.c_str();
|
|
62
|
-
}
|
|
63
|
-
|
|
64
|
-
static const char* IMG_BASE64_TAG_BEGIN = "<img src=\"data:image/jpeg;base64,";
|
|
65
|
-
static const char* IMG_BASE64_TAG_END = "\">";
|
|
66
|
-
|
|
67
|
-
static void find_image_tag_in_prompt(const std::string& prompt, size_t& begin_out, size_t& end_out) {
|
|
68
|
-
begin_out = prompt.find(IMG_BASE64_TAG_BEGIN);
|
|
69
|
-
end_out = prompt.find(IMG_BASE64_TAG_END, (begin_out == std::string::npos) ? 0UL : begin_out);
|
|
70
|
-
}
|
|
71
|
-
|
|
72
|
-
static bool prompt_contains_image(const std::string& prompt) {
|
|
73
|
-
size_t begin, end;
|
|
74
|
-
find_image_tag_in_prompt(prompt, begin, end);
|
|
75
|
-
return (begin != std::string::npos);
|
|
76
|
-
}
|
|
77
|
-
|
|
78
|
-
// replaces the base64 image tag in the prompt with `replacement`
|
|
79
|
-
static llava_image_embed * llava_image_embed_make_with_prompt_base64(struct clip_ctx * ctx_clip, int n_threads, const std::string& prompt) {
|
|
80
|
-
size_t img_base64_str_start, img_base64_str_end;
|
|
81
|
-
find_image_tag_in_prompt(prompt, img_base64_str_start, img_base64_str_end);
|
|
82
|
-
if (img_base64_str_start == std::string::npos || img_base64_str_end == std::string::npos) {
|
|
83
|
-
LOG_ERR("%s: invalid base64 image tag. must be %s<base64 byte string>%s\n", __func__, IMG_BASE64_TAG_BEGIN, IMG_BASE64_TAG_END);
|
|
84
|
-
return NULL;
|
|
85
|
-
}
|
|
86
|
-
|
|
87
|
-
auto base64_bytes_start = img_base64_str_start + strlen(IMG_BASE64_TAG_BEGIN);
|
|
88
|
-
auto base64_bytes_count = img_base64_str_end - base64_bytes_start;
|
|
89
|
-
auto base64_str = prompt.substr(base64_bytes_start, base64_bytes_count );
|
|
90
|
-
|
|
91
|
-
auto required_bytes = base64::required_encode_size(base64_str.size());
|
|
92
|
-
auto img_bytes = std::vector<unsigned char>(required_bytes);
|
|
93
|
-
base64::decode(base64_str.begin(), base64_str.end(), img_bytes.begin());
|
|
94
|
-
|
|
95
|
-
auto embed = llava_image_embed_make_with_bytes(ctx_clip, n_threads, img_bytes.data(), img_bytes.size());
|
|
96
|
-
if (!embed) {
|
|
97
|
-
LOG_ERR("%s: could not load image from base64 string.\n", __func__);
|
|
98
|
-
return NULL;
|
|
99
|
-
}
|
|
100
|
-
|
|
101
|
-
return embed;
|
|
102
|
-
}
|
|
103
|
-
|
|
104
|
-
static std::string remove_image_from_prompt(const std::string& prompt, const char * replacement = "") {
|
|
105
|
-
size_t begin, end;
|
|
106
|
-
find_image_tag_in_prompt(prompt, begin, end);
|
|
107
|
-
if (begin == std::string::npos || end == std::string::npos) {
|
|
108
|
-
return prompt;
|
|
109
|
-
}
|
|
110
|
-
auto pre = prompt.substr(0, begin);
|
|
111
|
-
auto post = prompt.substr(end + strlen(IMG_BASE64_TAG_END));
|
|
112
|
-
return pre + replacement + post;
|
|
113
|
-
}
|
|
114
|
-
|
|
115
|
-
struct llava_context {
|
|
116
|
-
struct clip_ctx * ctx_clip = NULL;
|
|
117
|
-
struct llama_context * ctx_llama = NULL;
|
|
118
|
-
struct llama_model * model = NULL;
|
|
119
|
-
};
|
|
120
|
-
|
|
121
|
-
static void print_usage(int, char ** argv) {
|
|
122
|
-
LOG("\n example usage:\n");
|
|
123
|
-
LOG("\n %s -m <llava-v1.5-7b/ggml-model-q5_k.gguf> --mmproj <llava-v1.5-7b/mmproj-model-f16.gguf> --image <path/to/an/image.jpg> --image <path/to/another/image.jpg> [--temp 0.1] [-p \"describe the image in detail.\"]\n", argv[0]);
|
|
124
|
-
LOG("\n note: a lower temperature value like 0.1 is recommended for better quality.\n");
|
|
125
|
-
}
|
|
126
|
-
|
|
127
|
-
static struct llava_image_embed * load_image(llava_context * ctx_llava, common_params * params, const std::string & fname) {
|
|
128
|
-
|
|
129
|
-
// load and preprocess the image
|
|
130
|
-
llava_image_embed * embed = NULL;
|
|
131
|
-
auto prompt = params->prompt;
|
|
132
|
-
if (prompt_contains_image(prompt)) {
|
|
133
|
-
if (!params->image.empty()) {
|
|
134
|
-
LOG_INF("using base64 encoded image instead of command line image path\n");
|
|
135
|
-
}
|
|
136
|
-
embed = llava_image_embed_make_with_prompt_base64(ctx_llava->ctx_clip, params->cpuparams.n_threads, prompt);
|
|
137
|
-
if (!embed) {
|
|
138
|
-
LOG_ERR("%s: can't load image from prompt\n", __func__);
|
|
139
|
-
return NULL;
|
|
140
|
-
}
|
|
141
|
-
params->prompt = remove_image_from_prompt(prompt);
|
|
142
|
-
} else {
|
|
143
|
-
embed = llava_image_embed_make_with_filename(ctx_llava->ctx_clip, params->cpuparams.n_threads, fname.c_str());
|
|
144
|
-
if (!embed) {
|
|
145
|
-
fprintf(stderr, "%s: is %s really an image file?\n", __func__, fname.c_str());
|
|
146
|
-
return NULL;
|
|
147
|
-
}
|
|
148
|
-
}
|
|
149
|
-
|
|
150
|
-
return embed;
|
|
151
|
-
}
|
|
152
|
-
|
|
153
|
-
static void process_prompt(struct llava_context * ctx_llava, struct llava_image_embed * image_embed, common_params * params, const std::string & prompt) {
|
|
154
|
-
int n_past = 0;
|
|
155
|
-
|
|
156
|
-
const int max_tgt_len = params->n_predict < 0 ? 256 : params->n_predict;
|
|
157
|
-
|
|
158
|
-
std::string system_prompt, user_prompt;
|
|
159
|
-
size_t image_pos = prompt.find("<image>");
|
|
160
|
-
if (image_pos != std::string::npos) {
|
|
161
|
-
// new templating mode: Provide the full prompt including system message and use <image> as a placeholder for the image
|
|
162
|
-
system_prompt = prompt.substr(0, image_pos);
|
|
163
|
-
user_prompt = prompt.substr(image_pos + std::string("<image>").length());
|
|
164
|
-
LOG_INF("system_prompt: %s\n", system_prompt.c_str());
|
|
165
|
-
if (params->verbose_prompt) {
|
|
166
|
-
auto tmp = common_tokenize(ctx_llava->ctx_llama, system_prompt, true, true);
|
|
167
|
-
for (int i = 0; i < (int) tmp.size(); i++) {
|
|
168
|
-
LOG_INF("%6d -> '%s'\n", tmp[i], common_token_to_piece(ctx_llava->ctx_llama, tmp[i]).c_str());
|
|
169
|
-
}
|
|
170
|
-
}
|
|
171
|
-
LOG_INF("user_prompt: %s\n", user_prompt.c_str());
|
|
172
|
-
if (params->verbose_prompt) {
|
|
173
|
-
auto tmp = common_tokenize(ctx_llava->ctx_llama, user_prompt, true, true);
|
|
174
|
-
for (int i = 0; i < (int) tmp.size(); i++) {
|
|
175
|
-
LOG_INF("%6d -> '%s'\n", tmp[i], common_token_to_piece(ctx_llava->ctx_llama, tmp[i]).c_str());
|
|
176
|
-
}
|
|
177
|
-
}
|
|
178
|
-
} else {
|
|
179
|
-
// llava-1.5 native mode
|
|
180
|
-
system_prompt = "A chat between a curious human and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the human's questions.\nUSER:";
|
|
181
|
-
user_prompt = prompt + "\nASSISTANT:";
|
|
182
|
-
if (params->verbose_prompt) {
|
|
183
|
-
auto tmp = common_tokenize(ctx_llava->ctx_llama, user_prompt, true, true);
|
|
184
|
-
for (int i = 0; i < (int) tmp.size(); i++) {
|
|
185
|
-
LOG_INF("%6d -> '%s'\n", tmp[i], common_token_to_piece(ctx_llava->ctx_llama, tmp[i]).c_str());
|
|
186
|
-
}
|
|
187
|
-
}
|
|
188
|
-
}
|
|
189
|
-
|
|
190
|
-
eval_string(ctx_llava->ctx_llama, system_prompt.c_str(), params->n_batch, &n_past, true);
|
|
191
|
-
llava_eval_image_embed(ctx_llava->ctx_llama, image_embed, params->n_batch, &n_past);
|
|
192
|
-
eval_string(ctx_llava->ctx_llama, user_prompt.c_str(), params->n_batch, &n_past, false);
|
|
193
|
-
|
|
194
|
-
// generate the response
|
|
195
|
-
|
|
196
|
-
LOG("\n");
|
|
197
|
-
|
|
198
|
-
struct common_sampler * smpl = common_sampler_init(ctx_llava->model, params->sampling);
|
|
199
|
-
if (!smpl) {
|
|
200
|
-
LOG_ERR("%s: failed to initialize sampling subsystem\n", __func__);
|
|
201
|
-
exit(1);
|
|
202
|
-
}
|
|
203
|
-
|
|
204
|
-
std::string response = "";
|
|
205
|
-
for (int i = 0; i < max_tgt_len; i++) {
|
|
206
|
-
const char * tmp = sample(smpl, ctx_llava->ctx_llama, &n_past);
|
|
207
|
-
response += tmp;
|
|
208
|
-
if (strcmp(tmp, "</s>") == 0) break;
|
|
209
|
-
if (strstr(tmp, "###")) break; // Yi-VL behavior
|
|
210
|
-
LOG("%s", tmp);
|
|
211
|
-
if (strstr(response.c_str(), "<|im_end|>")) break; // Yi-34B llava-1.6 - for some reason those decode not as the correct token (tokenizer works)
|
|
212
|
-
if (strstr(response.c_str(), "<|im_start|>")) break; // Yi-34B llava-1.6
|
|
213
|
-
if (strstr(response.c_str(), "USER:")) break; // mistral llava-1.6
|
|
214
|
-
|
|
215
|
-
fflush(stdout);
|
|
216
|
-
}
|
|
217
|
-
|
|
218
|
-
common_sampler_free(smpl);
|
|
219
|
-
LOG("\n");
|
|
220
|
-
}
|
|
221
|
-
|
|
222
|
-
static struct llama_model * llava_init(common_params * params) {
|
|
223
|
-
llama_backend_init();
|
|
224
|
-
llama_numa_init(params->numa);
|
|
225
|
-
|
|
226
|
-
llama_model_params model_params = common_model_params_to_llama(*params);
|
|
227
|
-
|
|
228
|
-
llama_model * model = llama_model_load_from_file(params->model.c_str(), model_params);
|
|
229
|
-
if (model == NULL) {
|
|
230
|
-
LOG_ERR("%s: unable to load model\n" , __func__);
|
|
231
|
-
return NULL;
|
|
232
|
-
}
|
|
233
|
-
return model;
|
|
234
|
-
}
|
|
235
|
-
|
|
236
|
-
static struct llava_context * llava_init_context(common_params * params, llama_model * model) {
|
|
237
|
-
const char * clip_path = params->mmproj.c_str();
|
|
238
|
-
|
|
239
|
-
auto prompt = params->prompt;
|
|
240
|
-
if (prompt.empty()) {
|
|
241
|
-
prompt = "describe the image in detail.";
|
|
242
|
-
}
|
|
243
|
-
|
|
244
|
-
auto ctx_clip = clip_model_load(clip_path, /*verbosity=*/ 1);
|
|
245
|
-
|
|
246
|
-
llama_context_params ctx_params = common_context_params_to_llama(*params);
|
|
247
|
-
ctx_params.n_ctx = params->n_ctx < 2048 ? 2048 : params->n_ctx; // we need a longer context size to process image embeddings
|
|
248
|
-
|
|
249
|
-
llama_context * ctx_llama = llama_init_from_model(model, ctx_params);
|
|
250
|
-
|
|
251
|
-
if (ctx_llama == NULL) {
|
|
252
|
-
LOG_ERR("%s: failed to create the llama_context\n" , __func__);
|
|
253
|
-
return NULL;
|
|
254
|
-
}
|
|
255
|
-
|
|
256
|
-
auto * ctx_llava = (struct llava_context *)malloc(sizeof(llava_context));
|
|
257
|
-
|
|
258
|
-
ctx_llava->ctx_llama = ctx_llama;
|
|
259
|
-
ctx_llava->ctx_clip = ctx_clip;
|
|
260
|
-
ctx_llava->model = model;
|
|
261
|
-
return ctx_llava;
|
|
262
|
-
}
|
|
263
|
-
|
|
264
|
-
static void llava_free(struct llava_context * ctx_llava) {
|
|
265
|
-
if (ctx_llava->ctx_clip) {
|
|
266
|
-
clip_free(ctx_llava->ctx_clip);
|
|
267
|
-
ctx_llava->ctx_clip = NULL;
|
|
268
|
-
}
|
|
269
|
-
|
|
270
|
-
llama_free(ctx_llava->ctx_llama);
|
|
271
|
-
llama_model_free(ctx_llava->model);
|
|
272
|
-
llama_backend_free();
|
|
273
|
-
}
|
|
274
|
-
|
|
275
|
-
int main(int argc, char ** argv) {
|
|
276
|
-
ggml_time_init();
|
|
277
|
-
|
|
278
|
-
common_params params;
|
|
279
|
-
|
|
280
|
-
if (!common_params_parse(argc, argv, params, LLAMA_EXAMPLE_LLAVA, print_usage)) {
|
|
281
|
-
return 1;
|
|
282
|
-
}
|
|
283
|
-
|
|
284
|
-
common_init();
|
|
285
|
-
|
|
286
|
-
if (params.mmproj.empty() || (params.image.empty() && !prompt_contains_image(params.prompt))) {
|
|
287
|
-
print_usage(argc, argv);
|
|
288
|
-
return 1;
|
|
289
|
-
}
|
|
290
|
-
|
|
291
|
-
auto * model = llava_init(¶ms);
|
|
292
|
-
if (model == NULL) {
|
|
293
|
-
fprintf(stderr, "%s: error: failed to init llava model\n", __func__);
|
|
294
|
-
return 1;
|
|
295
|
-
}
|
|
296
|
-
|
|
297
|
-
if (prompt_contains_image(params.prompt)) {
|
|
298
|
-
auto * ctx_llava = llava_init_context(¶ms, model);
|
|
299
|
-
|
|
300
|
-
auto * image_embed = load_image(ctx_llava, ¶ms, "");
|
|
301
|
-
|
|
302
|
-
// process the prompt
|
|
303
|
-
process_prompt(ctx_llava, image_embed, ¶ms, params.prompt);
|
|
304
|
-
|
|
305
|
-
llama_perf_context_print(ctx_llava->ctx_llama);
|
|
306
|
-
llava_image_embed_free(image_embed);
|
|
307
|
-
ctx_llava->model = NULL;
|
|
308
|
-
llava_free(ctx_llava);
|
|
309
|
-
} else {
|
|
310
|
-
for (auto & image : params.image) {
|
|
311
|
-
auto * ctx_llava = llava_init_context(¶ms, model);
|
|
312
|
-
|
|
313
|
-
auto * image_embed = load_image(ctx_llava, ¶ms, image);
|
|
314
|
-
if (!image_embed) {
|
|
315
|
-
LOG_ERR("%s: failed to load image %s. Terminating\n\n", __func__, image.c_str());
|
|
316
|
-
return 1;
|
|
317
|
-
}
|
|
318
|
-
|
|
319
|
-
// process the prompt
|
|
320
|
-
process_prompt(ctx_llava, image_embed, ¶ms, params.prompt);
|
|
321
|
-
|
|
322
|
-
llama_perf_context_print(ctx_llava->ctx_llama);
|
|
323
|
-
llava_image_embed_free(image_embed);
|
|
324
|
-
ctx_llava->model = NULL;
|
|
325
|
-
llava_free(ctx_llava);
|
|
326
|
-
}
|
|
327
|
-
}
|
|
328
|
-
|
|
329
|
-
llama_model_free(model);
|
|
330
|
-
|
|
331
|
-
return 0;
|
|
332
|
-
}
|
|
@@ -1,354 +0,0 @@
|
|
|
1
|
-
#include "arg.h"
|
|
2
|
-
#include "log.h"
|
|
3
|
-
#include "common.h"
|
|
4
|
-
#include "sampling.h"
|
|
5
|
-
#include "clip.h"
|
|
6
|
-
#include "llava.h"
|
|
7
|
-
#include "llama.h"
|
|
8
|
-
#include "ggml.h"
|
|
9
|
-
|
|
10
|
-
#include <algorithm>
|
|
11
|
-
#include <cstdio>
|
|
12
|
-
#include <cstdlib>
|
|
13
|
-
#include <cstring>
|
|
14
|
-
#include <vector>
|
|
15
|
-
#include <iostream> // TODO: remove me
|
|
16
|
-
|
|
17
|
-
struct llava_context {
|
|
18
|
-
struct clip_ctx * ctx_clip = NULL;
|
|
19
|
-
struct llama_context * ctx_llama = NULL;
|
|
20
|
-
struct llama_model * model = NULL;
|
|
21
|
-
};
|
|
22
|
-
|
|
23
|
-
static void show_additional_info(int /*argc*/, char ** argv) {
|
|
24
|
-
LOG("\nexample usage:\n\n%s -m <llava-v1.5-7b/ggml-model-q5_k.gguf> --mmproj <llava-v1.5-7b/mmproj-model-f16.gguf> --image <path/to/an/image.jpg> --image <path/to/another/image.jpg> [--temp 0.1] [-p \"describe the image in detail.\"]\n", argv[0]);
|
|
25
|
-
LOG("\nnote: a lower temperature value like 0.1 is recommended for better quality.\n");
|
|
26
|
-
}
|
|
27
|
-
|
|
28
|
-
static struct llama_model * llava_init(common_params * params) {
|
|
29
|
-
llama_backend_init();
|
|
30
|
-
llama_numa_init(params->numa);
|
|
31
|
-
|
|
32
|
-
llama_model_params model_params = common_model_params_to_llama(*params);
|
|
33
|
-
|
|
34
|
-
llama_model * model = llama_model_load_from_file(params->model.c_str(), model_params);
|
|
35
|
-
if (model == NULL) {
|
|
36
|
-
LOG_ERR("%s: unable to load model\n" , __func__);
|
|
37
|
-
return NULL;
|
|
38
|
-
}
|
|
39
|
-
return model;
|
|
40
|
-
}
|
|
41
|
-
|
|
42
|
-
static struct llava_context * llava_init_context(common_params * params, llama_model * model) {
|
|
43
|
-
auto prompt = params->prompt;
|
|
44
|
-
if (prompt.empty()) {
|
|
45
|
-
prompt = "describe the image in detail.";
|
|
46
|
-
}
|
|
47
|
-
|
|
48
|
-
llama_context_params ctx_params = common_context_params_to_llama(*params);
|
|
49
|
-
if (params->n_ctx < 2048) {
|
|
50
|
-
// warn user here, "Image processing requires at least 2048 context, setting context to 2048"
|
|
51
|
-
LOG_WRN("%s: Image processing requires at least 2048 context, setting context to 2048\n" , __func__);
|
|
52
|
-
ctx_params.n_ctx = 2048;
|
|
53
|
-
} else {
|
|
54
|
-
ctx_params.n_ctx = params->n_ctx;
|
|
55
|
-
}
|
|
56
|
-
|
|
57
|
-
llama_context * ctx_llama = llama_init_from_model(model, ctx_params);
|
|
58
|
-
|
|
59
|
-
if (ctx_llama == NULL) {
|
|
60
|
-
LOG_ERR("%s: failed to create the llama_context\n" , __func__);
|
|
61
|
-
return NULL;
|
|
62
|
-
}
|
|
63
|
-
|
|
64
|
-
auto * ctx_llava = (struct llava_context *)malloc(sizeof(llava_context));
|
|
65
|
-
|
|
66
|
-
ctx_llava->ctx_llama = ctx_llama;
|
|
67
|
-
ctx_llava->model = model;
|
|
68
|
-
return ctx_llava;
|
|
69
|
-
}
|
|
70
|
-
|
|
71
|
-
static void llava_free(struct llava_context * ctx_llava) {
|
|
72
|
-
if (ctx_llava->ctx_clip) {
|
|
73
|
-
clip_free(ctx_llava->ctx_clip);
|
|
74
|
-
ctx_llava->ctx_clip = NULL;
|
|
75
|
-
}
|
|
76
|
-
|
|
77
|
-
llama_free(ctx_llava->ctx_llama);
|
|
78
|
-
llama_model_free(ctx_llava->model);
|
|
79
|
-
llama_backend_free();
|
|
80
|
-
}
|
|
81
|
-
|
|
82
|
-
static struct clip_ctx * clip_init_context(common_params * params) {
|
|
83
|
-
const char * clip_path = params->mmproj.c_str();
|
|
84
|
-
|
|
85
|
-
auto prompt = params->prompt;
|
|
86
|
-
if (prompt.empty()) {
|
|
87
|
-
prompt = "describe the image in detail.";
|
|
88
|
-
}
|
|
89
|
-
struct clip_context_params clip_params = {
|
|
90
|
-
/* use_gpu */ params->n_gpu_layers != 0,
|
|
91
|
-
/* verbosity */ params->verbosity,
|
|
92
|
-
};
|
|
93
|
-
auto * ctx_clip = clip_init(clip_path, clip_params);
|
|
94
|
-
return ctx_clip;
|
|
95
|
-
}
|
|
96
|
-
|
|
97
|
-
static bool eval_tokens(struct llama_context * ctx_llama, std::vector<llama_token> tokens, int n_batch, int * n_past) {
|
|
98
|
-
int N = (int) tokens.size();
|
|
99
|
-
for (int i = 0; i < N; i += n_batch) {
|
|
100
|
-
int n_eval = (int) tokens.size() - i;
|
|
101
|
-
if (n_eval > n_batch) {
|
|
102
|
-
n_eval = n_batch;
|
|
103
|
-
}
|
|
104
|
-
if (llama_decode(ctx_llama, llama_batch_get_one(&tokens[i], n_eval))) {
|
|
105
|
-
LOG_ERR("%s : failed to eval. token %d/%d (batch size %d, n_past %d)\n", __func__, i, N, n_batch, *n_past);
|
|
106
|
-
return false;
|
|
107
|
-
}
|
|
108
|
-
*n_past += n_eval;
|
|
109
|
-
}
|
|
110
|
-
return true;
|
|
111
|
-
}
|
|
112
|
-
|
|
113
|
-
static bool eval_id(struct llama_context * ctx_llama, int id, int * n_past) {
|
|
114
|
-
std::vector<llama_token> tokens;
|
|
115
|
-
tokens.push_back(id);
|
|
116
|
-
return eval_tokens(ctx_llama, tokens, 1, n_past);
|
|
117
|
-
}
|
|
118
|
-
|
|
119
|
-
static bool eval_string(struct llama_context * ctx_llama, const char* str, int n_batch, int * n_past, bool add_bos){
|
|
120
|
-
std::string str2 = str;
|
|
121
|
-
std::vector<llama_token> embd_inp = common_tokenize(ctx_llama, str2, add_bos, true);
|
|
122
|
-
return eval_tokens(ctx_llama, embd_inp, n_batch, n_past);
|
|
123
|
-
}
|
|
124
|
-
|
|
125
|
-
static void process_eval_image_embed(struct llava_context * ctx_llava, const struct llava_image_embed * embeds, int n_batch, int * n_past, int idx) {
|
|
126
|
-
float * image_embed = (float *)malloc(clip_embd_nbytes(ctx_llava->ctx_clip));
|
|
127
|
-
std::memcpy(image_embed, embeds->embed + idx * clip_n_patches(ctx_llava->ctx_clip) * clip_n_mmproj_embd(ctx_llava->ctx_clip), clip_embd_nbytes(ctx_llava->ctx_clip));
|
|
128
|
-
|
|
129
|
-
auto * slice_embed = (llava_image_embed*)malloc(sizeof(llava_image_embed));
|
|
130
|
-
slice_embed->embed = image_embed;
|
|
131
|
-
slice_embed->n_image_pos = clip_n_patches(ctx_llava->ctx_clip);
|
|
132
|
-
llava_eval_image_embed(ctx_llava->ctx_llama, slice_embed, n_batch, n_past);
|
|
133
|
-
llava_image_embed_free(slice_embed);
|
|
134
|
-
}
|
|
135
|
-
|
|
136
|
-
static void process_image(struct llava_context * ctx_llava, struct llava_image_embed * embeds, common_params * params, int &n_past) {
|
|
137
|
-
std::string system_prompt;
|
|
138
|
-
int idx = 0;
|
|
139
|
-
int num_image_embeds = embeds->n_image_pos / clip_n_patches(ctx_llava->ctx_clip);
|
|
140
|
-
int has_minicpmv_projector = clip_is_minicpmv(ctx_llava->ctx_clip);
|
|
141
|
-
if (has_minicpmv_projector == 2) {
|
|
142
|
-
system_prompt = "<|begin_of_text|><|start_header_id|>user<|end_header_id|>\n\n";
|
|
143
|
-
}
|
|
144
|
-
else if (has_minicpmv_projector == 3) {
|
|
145
|
-
system_prompt = "<|im_start|>user\n";
|
|
146
|
-
}
|
|
147
|
-
else if (has_minicpmv_projector == 4) {
|
|
148
|
-
system_prompt = "<|im_start|>user\n";
|
|
149
|
-
}
|
|
150
|
-
LOG_INF("%s: image token past: %d\n", __func__, n_past);
|
|
151
|
-
eval_string(ctx_llava->ctx_llama, (system_prompt+"<image>").c_str(), params->n_batch, &n_past, false);
|
|
152
|
-
process_eval_image_embed(ctx_llava, embeds, params->n_batch, &n_past, idx++);
|
|
153
|
-
eval_string(ctx_llava->ctx_llama, std::string("</image>").c_str(), params->n_batch, &n_past, false);
|
|
154
|
-
if (num_image_embeds > 1) {
|
|
155
|
-
if (has_minicpmv_projector == 2) {
|
|
156
|
-
size_t num_image_embeds_col = clip_uhd_num_image_embeds_col(ctx_llava->ctx_clip);
|
|
157
|
-
eval_string(ctx_llava->ctx_llama, std::string("<slice>").c_str(), params->n_batch, &n_past, false);
|
|
158
|
-
for (size_t i = 0; i < (num_image_embeds-1)/num_image_embeds_col; ++i) {
|
|
159
|
-
for (size_t j = 0; j < num_image_embeds_col; ++j) {
|
|
160
|
-
eval_string(ctx_llava->ctx_llama, std::string("<image>").c_str(), params->n_batch, &n_past, false);
|
|
161
|
-
process_eval_image_embed(ctx_llava, embeds, params->n_batch, &n_past, idx++);
|
|
162
|
-
eval_string(ctx_llava->ctx_llama, std::string("</image>").c_str(), params->n_batch, &n_past, false);
|
|
163
|
-
if (j == num_image_embeds_col - 1) {
|
|
164
|
-
eval_string(ctx_llava->ctx_llama, std::string("\n").c_str(), params->n_batch, &n_past, false);
|
|
165
|
-
}
|
|
166
|
-
}
|
|
167
|
-
}
|
|
168
|
-
eval_string(ctx_llava->ctx_llama, std::string("</slice>").c_str(), params->n_batch, &n_past, false);
|
|
169
|
-
}
|
|
170
|
-
else if (has_minicpmv_projector == 3 || has_minicpmv_projector == 4) {
|
|
171
|
-
size_t num_image_embeds_col = clip_uhd_num_image_embeds_col(ctx_llava->ctx_clip);
|
|
172
|
-
for (size_t i = 0; i < (num_image_embeds-1)/num_image_embeds_col; ++i) {
|
|
173
|
-
for (size_t j = 0; j < num_image_embeds_col; ++j) {
|
|
174
|
-
eval_string(ctx_llava->ctx_llama, std::string("<slice>").c_str(), params->n_batch, &n_past, false);
|
|
175
|
-
process_eval_image_embed(ctx_llava, embeds, params->n_batch, &n_past, idx++);
|
|
176
|
-
eval_string(ctx_llava->ctx_llama, std::string("</slice>").c_str(), params->n_batch, &n_past, false);
|
|
177
|
-
if (j == num_image_embeds_col - 1) {
|
|
178
|
-
eval_string(ctx_llava->ctx_llama, std::string("\n").c_str(), params->n_batch, &n_past, false);
|
|
179
|
-
}
|
|
180
|
-
}
|
|
181
|
-
}
|
|
182
|
-
}
|
|
183
|
-
}
|
|
184
|
-
LOG_INF("%s: image token past: %d\n", __func__, n_past);
|
|
185
|
-
}
|
|
186
|
-
|
|
187
|
-
static const char * sample(struct common_sampler * smpl,
|
|
188
|
-
struct llama_context * ctx_llama,
|
|
189
|
-
int * n_past) {
|
|
190
|
-
const llama_token id = common_sampler_sample(smpl, ctx_llama, -1);
|
|
191
|
-
common_sampler_accept(smpl, id, true);
|
|
192
|
-
|
|
193
|
-
const llama_model * model = llama_get_model(ctx_llama);
|
|
194
|
-
const llama_vocab * vocab = llama_model_get_vocab(model);
|
|
195
|
-
|
|
196
|
-
static std::string ret;
|
|
197
|
-
if (llama_vocab_is_eog(vocab, id)) {
|
|
198
|
-
ret = "</s>";
|
|
199
|
-
} else {
|
|
200
|
-
ret = common_token_to_piece(ctx_llama, id);
|
|
201
|
-
}
|
|
202
|
-
eval_id(ctx_llama, id, n_past);
|
|
203
|
-
return ret.c_str();
|
|
204
|
-
}
|
|
205
|
-
|
|
206
|
-
static struct llava_context * minicpmv_init(common_params * params, const std::string & fname, int &n_past){
|
|
207
|
-
auto * ctx_clip = clip_init_context(params);
|
|
208
|
-
auto * embeds = llava_image_embed_make_with_filename(ctx_clip, params->cpuparams.n_threads, fname.c_str());
|
|
209
|
-
if (!embeds) {
|
|
210
|
-
LOG_ERR("failed to load image %s. Terminating\n\n", fname.c_str());
|
|
211
|
-
return NULL;
|
|
212
|
-
}
|
|
213
|
-
|
|
214
|
-
// process the prompt
|
|
215
|
-
if (params->prompt.empty() && params->interactive == false) {
|
|
216
|
-
LOG_ERR("prompt should be given or interactive mode should be on");
|
|
217
|
-
return NULL;
|
|
218
|
-
}
|
|
219
|
-
|
|
220
|
-
auto * model = llava_init(params);
|
|
221
|
-
if (model == NULL) {
|
|
222
|
-
fprintf(stderr, "%s: error: failed to init minicpmv model\n", __func__);
|
|
223
|
-
return NULL;
|
|
224
|
-
}
|
|
225
|
-
const int64_t t_llava_init_start_us = ggml_time_us();
|
|
226
|
-
auto * ctx_llava = llava_init_context(params, model);
|
|
227
|
-
ctx_llava->ctx_clip = ctx_clip;
|
|
228
|
-
const int64_t t_llava_init_end_us = ggml_time_us();
|
|
229
|
-
float t_llava_init_ms = (t_llava_init_end_us - t_llava_init_start_us) / 1000.0;
|
|
230
|
-
LOG_INF("%s: llava init in %8.2f ms.\n", __func__, t_llava_init_ms);
|
|
231
|
-
|
|
232
|
-
const int64_t t_process_image_start_us = ggml_time_us();
|
|
233
|
-
process_image(ctx_llava, embeds, params, n_past);
|
|
234
|
-
const int64_t t_process_image_end_us = ggml_time_us();
|
|
235
|
-
float t_process_image_ms = (t_process_image_end_us - t_process_image_start_us) / 1000.0;
|
|
236
|
-
LOG_INF("%s: llama process image in %8.2f ms.\n", __func__, t_process_image_ms);
|
|
237
|
-
|
|
238
|
-
llava_image_embed_free(embeds);
|
|
239
|
-
return ctx_llava;
|
|
240
|
-
}
|
|
241
|
-
|
|
242
|
-
static struct common_sampler * llama_init(struct llava_context * ctx_llava, common_params * params, const std::string & prompt, int & n_past, bool is_first = false){
|
|
243
|
-
std::string user_prompt = prompt;
|
|
244
|
-
int has_minicpmv_projector = clip_is_minicpmv(ctx_llava->ctx_clip);
|
|
245
|
-
if (!is_first) {
|
|
246
|
-
if (has_minicpmv_projector == 2) {
|
|
247
|
-
user_prompt = "<|begin_of_text|><|start_header_id|>user<|end_header_id|>\n\n" + prompt;
|
|
248
|
-
}
|
|
249
|
-
else if (has_minicpmv_projector == 3) {
|
|
250
|
-
user_prompt = "<|im_start|>user\n" + prompt;
|
|
251
|
-
}
|
|
252
|
-
else if (has_minicpmv_projector == 4) {
|
|
253
|
-
user_prompt = "<|im_start|>user\n" + prompt;
|
|
254
|
-
}
|
|
255
|
-
}
|
|
256
|
-
|
|
257
|
-
eval_string(ctx_llava->ctx_llama, user_prompt.c_str(), params->n_batch, &n_past, false);
|
|
258
|
-
if (has_minicpmv_projector == 2) {
|
|
259
|
-
eval_string(ctx_llava->ctx_llama, "<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n", params->n_batch, &n_past, false);
|
|
260
|
-
}
|
|
261
|
-
else if (has_minicpmv_projector == 3) {
|
|
262
|
-
eval_string(ctx_llava->ctx_llama, "<|im_end|><|im_start|>assistant\n", params->n_batch, &n_past, false);
|
|
263
|
-
}
|
|
264
|
-
else if (has_minicpmv_projector == 4) {
|
|
265
|
-
eval_string(ctx_llava->ctx_llama, "<|im_end|><|im_start|>assistant\n", params->n_batch, &n_past, false);
|
|
266
|
-
}
|
|
267
|
-
|
|
268
|
-
// generate the response
|
|
269
|
-
|
|
270
|
-
LOG_INF("\n");
|
|
271
|
-
|
|
272
|
-
struct common_sampler * smpl = common_sampler_init(ctx_llava->model, params->sampling);
|
|
273
|
-
return smpl;
|
|
274
|
-
}
|
|
275
|
-
|
|
276
|
-
static const char * llama_loop(struct llava_context * ctx_llava,struct common_sampler * smpl, int &n_past){
|
|
277
|
-
|
|
278
|
-
const char * tmp = sample(smpl, ctx_llava->ctx_llama, &n_past);
|
|
279
|
-
return tmp;
|
|
280
|
-
}
|
|
281
|
-
|
|
282
|
-
int main(int argc, char ** argv) {
|
|
283
|
-
ggml_time_init();
|
|
284
|
-
|
|
285
|
-
common_params params;
|
|
286
|
-
|
|
287
|
-
if (!common_params_parse(argc, argv, params, LLAMA_EXAMPLE_LLAVA, show_additional_info)) {
|
|
288
|
-
return 1;
|
|
289
|
-
}
|
|
290
|
-
|
|
291
|
-
common_init();
|
|
292
|
-
|
|
293
|
-
if (params.mmproj.empty() || (params.image.empty())) {
|
|
294
|
-
show_additional_info(argc, argv);
|
|
295
|
-
return 1;
|
|
296
|
-
}
|
|
297
|
-
|
|
298
|
-
for (auto & image : params.image) {
|
|
299
|
-
int n_past = 0;
|
|
300
|
-
auto * ctx_llava = minicpmv_init(¶ms, image, n_past);
|
|
301
|
-
|
|
302
|
-
if (!params.prompt.empty()) {
|
|
303
|
-
LOG("<user>%s\n", params.prompt.c_str());
|
|
304
|
-
LOG("<assistant>");
|
|
305
|
-
auto * smpl = llama_init(ctx_llava, ¶ms, params.prompt, n_past, true);
|
|
306
|
-
const int max_tgt_len = params.n_predict < 0 ? 256 : params.n_predict;
|
|
307
|
-
std::string response;
|
|
308
|
-
bool have_tmp = false;
|
|
309
|
-
for (int i = 0; i < max_tgt_len; i++) {
|
|
310
|
-
const auto * tmp = llama_loop(ctx_llava, smpl, n_past);
|
|
311
|
-
response += tmp;
|
|
312
|
-
if (strcmp(tmp, "</s>") == 0){
|
|
313
|
-
if (!have_tmp) {
|
|
314
|
-
continue;
|
|
315
|
-
}
|
|
316
|
-
break;
|
|
317
|
-
}
|
|
318
|
-
if (strstr(tmp, "###")) break; // Yi-VL behavior
|
|
319
|
-
have_tmp = true;
|
|
320
|
-
printf("%s", tmp);
|
|
321
|
-
if (strstr(response.c_str(), "<user>")) break; // minicpm-v
|
|
322
|
-
|
|
323
|
-
fflush(stdout);
|
|
324
|
-
}
|
|
325
|
-
common_sampler_free(smpl);
|
|
326
|
-
}else {
|
|
327
|
-
while (true) {
|
|
328
|
-
LOG("<user>");
|
|
329
|
-
std::string prompt;
|
|
330
|
-
std::getline(std::cin, prompt);
|
|
331
|
-
LOG("<assistant>");
|
|
332
|
-
auto * smpl = llama_init(ctx_llava, ¶ms, prompt, n_past, true);
|
|
333
|
-
const int max_tgt_len = params.n_predict < 0 ? 256 : params.n_predict;
|
|
334
|
-
std::string response;
|
|
335
|
-
for (int i = 0; i < max_tgt_len; i++) {
|
|
336
|
-
const auto * tmp = llama_loop(ctx_llava, smpl, n_past);
|
|
337
|
-
response += tmp;
|
|
338
|
-
if (strcmp(tmp, "</s>") == 0) break;
|
|
339
|
-
printf("%s", tmp);// mistral llava-1.6
|
|
340
|
-
if (strstr(response.c_str(), "<user>")) break; // minicpm-v
|
|
341
|
-
fflush(stdout);
|
|
342
|
-
}
|
|
343
|
-
common_sampler_free(smpl);
|
|
344
|
-
}
|
|
345
|
-
}
|
|
346
|
-
printf("\n");
|
|
347
|
-
llama_perf_context_print(ctx_llava->ctx_llama);
|
|
348
|
-
|
|
349
|
-
ctx_llava->model = NULL;
|
|
350
|
-
llava_free(ctx_llava);
|
|
351
|
-
}
|
|
352
|
-
|
|
353
|
-
return 0;
|
|
354
|
-
}
|
|
@@ -1,6 +0,0 @@
|
|
|
1
|
-
set(TARGET llama-quantize-stats)
|
|
2
|
-
add_executable(${TARGET} quantize-stats.cpp)
|
|
3
|
-
install(TARGETS ${TARGET} RUNTIME)
|
|
4
|
-
target_link_libraries(${TARGET} PRIVATE llama build_info ${CMAKE_THREAD_LIBS_INIT})
|
|
5
|
-
target_include_directories(${TARGET} PRIVATE ../../common)
|
|
6
|
-
target_compile_features(${TARGET} PRIVATE cxx_std_17)
|