@@ -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