File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -692,7 +692,7 @@ def export_from_model(
692692 # some model configs may have issues with loading without parameters initialization
693693 try :
694694 misplaced_generation_parameters = model .config ._get_non_default_generation_parameters ()
695- except KeyError :
695+ except ( KeyError , TypeError ) :
696696 misplaced_generation_parameters = {}
697697 if isinstance (model , GenerationMixin ) and len (misplaced_generation_parameters ) > 0 :
698698 logger .warning (
Original file line number Diff line number Diff line change @@ -139,7 +139,7 @@ def __init__(
139139 # some model configs may have issues with loading without parameters initialization
140140 try :
141141 misplaced_generation_parameters = self .config ._get_non_default_generation_parameters ()
142- except KeyError :
142+ except ( KeyError , TypeError ) :
143143 misplaced_generation_parameters = {}
144144 if len (misplaced_generation_parameters ) > 0 :
145145 logger .warning (
Original file line number Diff line number Diff line change @@ -87,7 +87,7 @@ def __init__(
8787 # some model configs may have issues with loading without parameters initialization
8888 try :
8989 misplaced_generation_parameters = self .config ._get_non_default_generation_parameters ()
90- except KeyError :
90+ except ( KeyError , TypeError ) :
9191 misplaced_generation_parameters = {}
9292 if len (misplaced_generation_parameters ) > 0 :
9393 logger .warning (
You can’t perform that action at this time.
0 commit comments