put back

mosaicml · Jul 25, 2024 · e4c9691 · e4c9691
1 parent aea32d6
commit e4c9691
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -483,8 +483,6 @@ def tensor_hook(
             if dist.get_global_rank() != 0:
                 state_dict = {}
             return state_dict
-
-        assert False
 
         hooks = []
         for _, module in state_dict_model.named_modules():