from local

This commit is contained in:
dohe0342 2023-01-03 17:49:51 +09:00
parent 305a7ddc6b
commit 86cc17624c
3 changed files with 1 additions and 2 deletions

View File

@ -1538,8 +1538,7 @@ def run_adapter(rank, world_size, args, wb=None):
adapter_names = []
adapter_param = []
for n, p in model.named_parameters():
print(n)
if 'adapters' in n:
if 'adapters' in n or 'joiner' in n or 'simple' in n:
adapter_names.append(n)
adapter_param.append(p)
#else: