From 082fe837e86b517cf2b1af884cd34021333ab25f Mon Sep 17 00:00:00 2001 From: Weiming Zhao Date: Mon, 8 Jan 2024 21:49:35 +0000 Subject: [PATCH] dummy test; not for merge --- src/transformers/modeling_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 05d74d654252..84983cb3ce25 100644 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -513,6 +513,7 @@ def load_state_dict(checkpoint_file: Union[str, os.PathLike]): is_deepspeed_zero3_enabled() and torch.distributed.is_initialized() and torch.distributed.get_rank() > 0 ) or (is_fsdp_enabled() and not is_local_dist_rank_0()): map_location = "meta" + assert(0) else: map_location = "cpu"