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(&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);
|
||||
return 1;
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue