dflash: refactor draft model conversion (#25110)

* dflash: refactor draft model conversion * apply fix for eagle3 convert
2026-06-28 20:31:48 +02:00
parent c818263f2a
commit fa72bc6826
5 changed files with 31 additions and 17 deletions
@@ -1283,6 +1283,11 @@ class TensorNameMap:
        MODEL_TENSOR.ENC_OUTPUT_NORM: (
            "encoder.final_layer_norm", # t5
            "layer_norm",               # neobert
+            "model.hidden_norm",        # dflash
+        ),
+
+        MODEL_TENSOR.FC: (
+            "model.fc", # dflash
        ),

        MODEL_TENSOR.CLS: (