lora : allow 1d tensors
This commit is contained in:
parent
93b42fc99b
commit
3289eb0cb3
1 changed files with 1 additions and 1 deletions
|
@ -6372,7 +6372,7 @@ static int llama_apply_lora_from_file_internal(
|
||||||
fin.read_raw(&name_len, sizeof(name_len));
|
fin.read_raw(&name_len, sizeof(name_len));
|
||||||
fin.read_raw(&ftype, sizeof(ftype));
|
fin.read_raw(&ftype, sizeof(ftype));
|
||||||
|
|
||||||
if (n_dims != 2) {
|
if (n_dims != 1 && n_dims != 2) {
|
||||||
LLAMA_LOG_ERROR("%s: unsupported tensor dimension %d\n", __func__, n_dims);
|
LLAMA_LOG_ERROR("%s: unsupported tensor dimension %d\n", __func__, n_dims);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue