File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments