chore: import upstream snapshot with attribution

2026-07-13 12:35:17 +08:00
commit 344816a5d8
136 changed files with 25044 additions and 0 deletions
@@ -0,0 +1,20 @@
+import torch.nn
+from modules.backbones.wavenet import WaveNet
+from modules.backbones.lynxnet import LYNXNet
+from modules.backbones.lynxnet2 import LYNXNet2
+from utils import filter_kwargs
+
+BACKBONES = {
+    'wavenet': WaveNet,
+    'lynxnet': LYNXNet,
+    'lynxnet2': LYNXNet2,
+}
+
+
+def build_backbone(
+        out_dims: int, num_feats: int,
+        backbone_type: str, backbone_args: dict
+) -> torch.nn.Module:
+    backbone = BACKBONES[backbone_type]
+    kwargs = filter_kwargs(backbone_args, backbone)
+    return BACKBONES[backbone_type](out_dims, num_feats, **kwargs)