from local

This commit is contained in:
dohe0342 2023-05-18 16:35:25 +09:00
parent 853dbda1ac
commit 80d05654c3
3 changed files with 1 additions and 1 deletions

View File

@ -1587,7 +1587,7 @@ def run_adapter(rank, world_size, args, wb=None):
adapter_names = [] adapter_names = []
adapter_param = [] adapter_param = []
for n, p in model.named_parameters(): for n, p in model.named_parameters():
if 'bias' in n: if 'k_proj.bias' in n:
adapter_names.append(n) adapter_names.append(n)
adapter_param.append(p) adapter_param.append(p)
else: else: