Skip to content

Commit e2feb69

Browse files
nv-guomingznv-guomingz
nv-guomingz
authored andcommitted
Fix starcoder model loading issue huggingface#37765
Signed-off-by: nv-guomingz <37257613+nv-guomingz@users.noreply.github.com>
1 parent b2d70e9 commit e2feb69

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/transformers/modeling_utils.py

+2
Original file line numberDiff line numberDiff line change
@@ -735,6 +735,8 @@ def _load_state_dict_into_meta_model(
735735
if param_name not in expected_keys:
736736
continue
737737

738+
if not param.dtype.is_floating_point:
739+
continue
738740
# we need to use serialized_param_name as file pointer is untouched
739741
if is_meta_state_dict:
740742
# This is the name of the parameter as it appears on disk file

0 commit comments

Comments
 (0)