Update resampler.py

#79
by shawony - opened
Files changed (1) hide show
  1. resampler.py +1 -1
resampler.py CHANGED
@@ -12,7 +12,7 @@ from torch.nn.init import trunc_normal_
12
  from torch.nn.init import constant_, xavier_normal_, xavier_uniform_
13
  from transformers import PreTrainedModel
14
  from transformers.integrations import is_deepspeed_zero3_enabled
15
-
16
  def get_2d_sincos_pos_embed(embed_dim, image_size):
17
  """
18
  image_size: image_size or (image_height, image_width)
 
12
  from torch.nn.init import constant_, xavier_normal_, xavier_uniform_
13
  from transformers import PreTrainedModel
14
  from transformers.integrations import is_deepspeed_zero3_enabled
15
+ from typing import List
16
  def get_2d_sincos_pos_embed(embed_dim, image_size):
17
  """
18
  image_size: image_size or (image_height, image_width)