Add switchable DIFF V1 attention

2026-08-21 11:48:16 +08:00
parent 75c9f06114
commit bf5cae8758
7 changed files with 453 additions and 4 deletions
--- a/eval_data.py
+++ b/eval_data.py
@@ -231,6 +231,7 @@ def build_model_from_dataset(
        dist_mode=str(cfg_get(args, cfg, "dist_mode", "exponential")),
        dropout=float(cfg_get(args, cfg, "dropout", 0.0)),
        model_architecture=model_architecture,
+        attention_type=str(cfg_get(args, cfg, "attention_type", "standard")),
    )