SkywalkerLu
/

TriStageHLA-IM

Model card Files Files and versions

SkywalkerLu commited on Dec 16, 2025

Commit

3026373

·

verified ·

1 Parent(s): f4f8c7b

Update modeling_transhla2.py

Files changed (1) hide show

modeling_transhla2.py +1 -1

modeling_transhla2.py CHANGED Viewed

@@ -18,7 +18,7 @@ class TransHLA2(PreTrainedModel):
         n_layers = 4
         n_head = 8
-        d_model = 480
         d_ff = 64
         cnn_num_channel = 256
         region_embedding_size = 3

         n_layers = 4
         n_head = 8
+        d_model = config.d_model
         d_ff = 64
         cnn_num_channel = 256
         region_embedding_size = 3