| 127 | } |
| 128 | |
| 129 | static bool run(llama_context * ctx, const common_params & params) { |
| 130 | const llama_model * model = llama_get_model(ctx); |
| 131 | const llama_vocab * vocab = llama_model_get_vocab(model); |
| 132 | |
| 133 | const bool add_bos = llama_vocab_get_add_bos(vocab); |
| 134 | |
| 135 | std::vector<llama_token> tokens = common_tokenize(ctx, params.prompt, add_bos); |
| 136 | |
| 137 | if (llama_decode(ctx, llama_batch_get_one(tokens.data(), tokens.size()))) { |
| 138 | LOG_ERR("%s : failed to eval\n", __func__); |
| 139 | return false; |
| 140 | } |
| 141 | |
| 142 | return true; |
| 143 | } |
| 144 | |
| 145 | int main(int argc, char ** argv) { |
| 146 | callback_data cb_data; |
no test coverage detected