update新结构

2026-01-26 18:27:41 +08:00
parent bc838d7cd7
commit cb610281ce
6 changed files with 55 additions and 2 deletions
--- a/example/export_samples.py
+++ b/example/export_samples.py
@@ -151,6 +151,9 @@ def main():
        ff_mult=int(cfg.get("model_ff_mult", 2)),
        pos_dim=int(cfg.get("model_pos_dim", 64)),
        use_pos_embed=bool(cfg.get("model_use_pos_embed", True)),
+        use_feature_graph=bool(cfg.get("model_use_feature_graph", False)),
+        feature_graph_scale=float(cfg.get("feature_graph_scale", 0.1)),
+        feature_graph_dropout=float(cfg.get("feature_graph_dropout", 0.0)),
        cond_vocab_size=cond_vocab_size if use_condition else 0,
        cond_dim=int(cfg.get("cond_dim", 32)),
        use_tanh_eps=bool(cfg.get("use_tanh_eps", False)),