Skip to content

Commit

Permalink
fix(cpp): options get lora_list
Browse files Browse the repository at this point in the history
  • Loading branch information
jhen0409 committed Jan 22, 2025
1 parent e1abacf commit 48bf8a5
Showing 1 changed file with 17 additions and 15 deletions.
32 changes: 17 additions & 15 deletions src/LlamaContext.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -229,22 +229,24 @@ LlamaContext::LlamaContext(const Napi::CallbackInfo &info)
lora.push_back(la);
}

auto lora_list = options.Get("lora_list").As<Napi::Array>();
if (lora_list != nullptr) {
int lora_list_size = lora_list.Length();
for (int i = 0; i < lora_list_size; i++) {
auto lora_adapter = lora_list.Get(i).As<Napi::Object>();
auto path = lora_adapter.Get("path").ToString();
if (path != nullptr) {
common_adapter_lora_info la;
la.path = path;
la.scale = lora_adapter.Get("scaled").ToNumber().FloatValue();
la.ptr = llama_adapter_lora_init(model, path.Utf8Value().c_str());
if (la.ptr == nullptr) {
Napi::TypeError::New(env, "Failed to load lora adapter")
.ThrowAsJavaScriptException();
if (options.Has("lora_list") && options.Get("lora_list").IsArray()) {
auto lora_list = options.Get("lora_list").As<Napi::Array>();
if (lora_list != nullptr) {
int lora_list_size = lora_list.Length();
for (int i = 0; i < lora_list_size; i++) {
auto lora_adapter = lora_list.Get(i).As<Napi::Object>();
auto path = lora_adapter.Get("path").ToString();
if (path != nullptr) {
common_adapter_lora_info la;
la.path = path;
la.scale = lora_adapter.Get("scaled").ToNumber().FloatValue();
la.ptr = llama_adapter_lora_init(model, path.Utf8Value().c_str());
if (la.ptr == nullptr) {
Napi::TypeError::New(env, "Failed to load lora adapter")
.ThrowAsJavaScriptException();
}
lora.push_back(la);
}
lora.push_back(la);
}
}
}
Expand Down

0 comments on commit 48bf8a5

Please sign in to comment.