wip refactor

utils.py

@@ -0,0 +1,7 @@
+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)