8 lines
219 B
Python
8 lines
219 B
Python
from peft.utils import _get_submodules
|
|
import torch
|
|
|
|
|
|
def replace_module(model, key: str, module: torch.nn.Module):
|
|
parent, target, target_name = _get_submodules(model, key)
|
|
setattr(parent, target_name, module)
|