Update model.py

model.py

@@ -9,6 +9,7 @@ from .utils import PatchEmbed, create_pretrained_model, repeat_token
 
 from einops import rearrange
 
+from typing import List 
 
 def conv3x3(in_channels, out_channels, stride=1):
     "3x3 convolution with padding"
@@ -26,7 +27,7 @@ class GRAMT(nn.Module):
         decoder_depth: int = 8,
         decoder_num_heads: int = 8,
         decoder_embedding_dim: int = 512,
-        decoder_window_sizes: list[int] = [2, 5, 10, 25, 50, 100, 0, 0],
+        decoder_window_sizes: List[int] = [2, 5, 10, 25, 50, 100, 0, 0],
         encoder_num_layers = 12,
         encoder_num_heads = 12,
         encoder_hidden_dim = 768,