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