Skip to content

Commit e4cdd82

Browse files
author
Felipe Mello
committed
add todo
1 parent e69fd47 commit e4cdd82

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torchtune/models/clip/_position_embeddings.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -282,6 +282,7 @@ def _resize_local_position_embedding(
282282

283283
return local_pos_embed
284284

285+
# TODO: Switchswitch to public method after 2.5 is stable
285286
@staticmethod
286287
def _resize_global_position_embedding(
287288
global_pos_embed: torch.Tensor,
@@ -467,6 +468,7 @@ def __init__(
467468
# Register load hook to interpolate positional embeddings
468469
self._register_load_state_dict_pre_hook(self._load_state_dict_hook)
469470

471+
# TODO: Switchswitch to public method after 2.5 is stable
470472
@torch.no_grad()
471473
def _load_state_dict_hook(
472474
self,

0 commit comments

Comments
 (0)