Skip to content

Commit e6730ce

Browse files
committed
move deprecate, make private
1 parent 017d4e9 commit e6730ce

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

src/diffusers/models/embeddings.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)