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,24 @@
+def get_backbone_type(root_config: dict, nested_config: dict = None):
+    if nested_config is None:
+        nested_config = root_config
+    return nested_config.get(
+        'backbone_type',
+        root_config.get(
+            'backbone_type',
+            root_config.get('diff_decoder_type', 'wavenet')
+        )
+    )
+
+
+def get_backbone_args(config: dict, backbone_type: str):
+    args = config.get('backbone_args')
+    if args is not None:
+        return args
+    elif backbone_type == 'wavenet':
+        return {
+            'num_layers': config.get('residual_layers'),
+            'num_channels': config.get('residual_channels'),
+            'dilation_cycle_length': config.get('dilation_cycle_length'),
+        }
+    else:
+        return None