Skip to content

Commit 781b49a

Browse files
committed
feat: add nEmbd in model details
1 parent 28906e6 commit 781b49a

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

android/src/main/jni.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -412,6 +412,7 @@ Java_com_rnllama_LlamaContext_loadModelDetails(
412412
llama_model_desc(llama->model, desc, sizeof(desc));
413413
putString(env, result, "desc", desc);
414414
putDouble(env, result, "size", llama_model_size(llama->model));
415+
putDouble(env, result, "nEmbd", llama_n_embd(llama->model));
415416
putDouble(env, result, "nParams", llama_model_n_params(llama->model));
416417
putBoolean(env, result, "isChatTemplateSupported", llama->validateModelChatTemplate());
417418
putMap(env, result, "metadata", meta);

ios/RNLlamaContext.mm

+1
Original file line numberDiff line numberDiff line change
@@ -221,6 +221,7 @@ - (NSDictionary *)modelInfo {
221221
return @{
222222
@"desc": [NSString stringWithUTF8String:desc],
223223
@"size": @(llama_model_size(llama->model)),
224+
@"nEmbd": @(llama_n_embd(llama->model)),
224225
@"nParams": @(llama_model_n_params(llama->model)),
225226
@"isChatTemplateSupported": @(llama->validateModelChatTemplate()),
226227
@"metadata": meta

0 commit comments

Comments
 (0)