Skip to content

Commit 96bd9e1

Browse files
committed
rebase
1 parent 0d4e554 commit 96bd9e1

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

tests/torchtune/training/checkpointing/test_distributed_checkpointer.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -156,16 +156,9 @@ def test_save_load_adapter_checkpoint(
156156
self, distributed_checkpointer, state_dict, empty_state_dict
157157
):
158158
"""
159-
<<<<<<< HEAD
160-
Test ``load_checkpoint`` method of an adapter checkpoint within the DistributedCheckpointer.
161-
=======
162-
Test ``load_checkpoint`` method within the DistributedCheckpointer.
163-
164-
We test:
165-
* ``load_checkpoint`` loads the right sets of keys
166-
* Internal state of the checkpointer is correctly updated.
167-
>>>>>>> c63bac3b (squashed commit)
159+
Test ``load_checkpoint`` method of an adapter checkpoint within the DistributedCheckpointer.
168160
"""
161+
169162
distributed_checkpointer.save_checkpoint(
170163
state_dict=state_dict, epoch=1, save_async=False, adapter_only=True
171164
)

0 commit comments

Comments
 (0)