mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
a21211327a
commit
f525a35a26
Binary file not shown.
@ -1596,7 +1596,7 @@ def run_adapter(rank, world_size, args, wb=None):
|
||||
for i, module in enumerate(model.modules()):
|
||||
if isinstance(module, fairseq.modules.multihead_attention.MultiheadAttention):
|
||||
for m in module.modules():
|
||||
lora_module.append(LoRAHook())
|
||||
lora_module.append(LoRAHook(m))
|
||||
|
||||
print('-'*30)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user