Skip to content

Commit 47e795b

Browse files
misc don't recreate it (#42394)
* misc don't recreate it * style --------- Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
1 parent b605555 commit 47e795b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/transformers/core_model_loading.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -317,8 +317,8 @@ def convert(
317317
config=None,
318318
hf_quantizer=None,
319319
missing_keys: Optional[MutableSet[str]] = None,
320+
misc: Optional[MutableMapping[str, str]] = None,
320321
):
321-
misc = {}
322322
for pattern, futures in self.collected_tensors.items():
323323
self.collected_tensors[pattern] = [future.result() for future in futures]
324324

@@ -358,8 +358,8 @@ def convert(
358358
config=None,
359359
hf_quantizer=None,
360360
missing_keys: Optional[MutableSet[str]] = None,
361+
misc: Optional[MutableMapping[str, str]] = None,
361362
):
362-
misc = {}
363363
for pattern, futures in self.collected_tensors.items():
364364
self.collected_tensors[pattern] = [future.result() for future in futures]
365365

@@ -782,6 +782,7 @@ def convert_and_load_state_dict_in_model(
782782
config=model.config,
783783
hf_quantizer=hf_quantizer,
784784
missing_keys=missing_keys,
785+
misc=misc,
785786
)
786787
for target_name, param in realized_value.items():
787788
param = param[0] if isinstance(param, list) else param

0 commit comments

Comments
 (0)