Lint

Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
pytorch · Nov 25, 2024 · 1a9bf51 · 1a9bf51
1 parent 9a1f4d9
commit 1a9bf51
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/torchtune/modules/_export/_position_embeddings.py b/torchtune/modules/_export/_position_embeddings.py
@@ -356,9 +356,9 @@ def _load_state_dict_hook(
                 )
 
             # update state dict
-            state_dict[prefix + "local_token_positional_embedding"] = (
-                inpt_local_pos_embed
-            )
+            state_dict[
+                prefix + "local_token_positional_embedding"
+            ] = inpt_local_pos_embed
             if (
                 inpt_local_pos_embed.shape
                 != self.local_token_positional_embedding.shape
@@ -418,9 +418,9 @@ def _load_state_dict_hook(
                 )
 
             # update state dict
-            state_dict[prefix + "global_token_positional_embedding"] = (
-                inpt_global_pos_embed
-            )
+            state_dict[
+                prefix + "global_token_positional_embedding"
+            ] = inpt_global_pos_embed
             if (
                 inpt_global_pos_embed.shape
                 != self.global_token_positional_embedding.shape