From d0e4630dd6c412a9cf4e2413606ed4d061508636 Mon Sep 17 00:00:00 2001 From: thanhvc3 Date: Sun, 28 Apr 2024 15:26:24 +0700 Subject: [PATCH] try gtp vit --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 464c74f..598fb5c 100644 --- 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