try gtp vit

2024-04-28 15:26:24 +07:00 · 2024-04-28 15:26:24 +07:00 · d0e4630dd6
commit d0e4630dd6
parent 08a3780ba6
1 changed files with 1 additions and 1 deletions
--- a/models.py
+++ b/models.py
@ -758,7 +758,7 @@ def basic_blocks(dim, index, layers,
            use_layer_scale=use_layer_scale, 
            layer_scale_init_value=layer_scale_init_value, 
            ))
-    blocks = SeqModel(blocks)
+    blocks = SeqModel(*blocks)

    return blocks