Skip to content

Commit 9b83493

Browse files
committed
use n_gpu_layers param
1 parent bf1d72b commit 9b83493

File tree

3 files changed

+7
-3
lines changed

3 files changed

+7
-3
lines changed

examples/llava/clip.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1209,7 +1209,7 @@ struct clip_ctx * clip_model_load(const char * fname, const int verbosity = 1) {
12091209
});
12101210
}
12111211

1212-
struct clip_ctx * clip_init(const char * fname, clip_context_params ctx_params) {
1212+
struct clip_ctx * clip_init(const char * fname, struct clip_context_params ctx_params) {
12131213
int verbosity = ctx_params.verbosity;
12141214
struct ggml_context * meta = NULL;
12151215

examples/llava/clip.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ struct clip_context_params {
4747
// deprecated, use clip_init
4848
CLIP_API struct clip_ctx * clip_model_load(const char * fname, int verbosity);
4949

50-
CLIP_API struct clip_ctx * clip_init(const char * fname, clip_context_params ctx_params);
50+
CLIP_API struct clip_ctx * clip_init(const char * fname, struct clip_context_params ctx_params);
5151

5252
CLIP_API void clip_free(struct clip_ctx * ctx);
5353

examples/llava/minicpmv-cli.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,11 @@ static struct clip_ctx * clip_init_context(common_params * params) {
8686
if (prompt.empty()) {
8787
prompt = "describe the image in detail.";
8888
}
89-
auto * ctx_clip = clip_model_load(clip_path, /*verbosity=*/ 1);
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);
9094
return ctx_clip;
9195
}
9296

0 commit comments

Comments
 (0)