@@ -109,13 +109,7 @@ def get_3d_sincos_pos_embed(
109109            embed_dim]`. 
110110    """ 
111111    if  output_type  ==  "np" :
112-         deprecation_message  =  (
113-             "`get_3d_sincos_pos_embed` uses `torch` and supports `device`." 
114-             " `from_numpy` is no longer required." 
115-             "  Pass `output_type='pt' to use the new version now." 
116-         )
117-         deprecate ("output_type=='np'" , "0.33.0" , deprecation_message , standard_warn = False )
118-         return  get_3d_sincos_pos_embed_np (
112+         return  _get_3d_sincos_pos_embed_np (
119113            embed_dim = embed_dim ,
120114            spatial_size = spatial_size ,
121115            temporal_size = temporal_size ,
@@ -156,7 +150,7 @@ def get_3d_sincos_pos_embed(
156150    return  pos_embed 
157151
158152
159- def  get_3d_sincos_pos_embed_np (
153+ def  _get_3d_sincos_pos_embed_np (
160154    embed_dim : int ,
161155    spatial_size : Union [int , Tuple [int , int ]],
162156    temporal_size : int ,
@@ -184,6 +178,12 @@ def get_3d_sincos_pos_embed_np(
184178            The 3D sinusoidal positional embeddings of shape `[temporal_size, spatial_size[0] * spatial_size[1], 
185179            embed_dim]`. 
186180    """ 
181+     deprecation_message  =  (
182+         "`get_3d_sincos_pos_embed` uses `torch` and supports `device`." 
183+         " `from_numpy` is no longer required." 
184+         "  Pass `output_type='pt' to use the new version now." 
185+     )
186+     deprecate ("output_type=='np'" , "0.33.0" , deprecation_message , standard_warn = False )
187187    if  embed_dim  %  4  !=  0 :
188188        raise  ValueError ("`embed_dim` must be divisible by 4" )
189189    if  isinstance (spatial_size , int ):
0 commit comments