Skip to content

Commit 34b6e24

Browse files
committed
fix saving model w/o default config
1 parent 016d538 commit 34b6e24

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

src/transformers/configuration_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1033,6 +1033,7 @@ def _get_non_default_generation_parameters(self) -> Dict[str, Any]:
10331033
if decoder_config is not self:
10341034
default_config = decoder_config.__class__()
10351035
else:
1036+
default_config = None
10361037
decoder_config = None
10371038

10381039
# If it is a composite model, we want to check the subconfig that will be used for generation

src/transformers/models/moshi/convert_moshi_transformers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,8 @@ def convert_checkpoint(
223223
original_checkpoint.update({f"audio_encoder.{key}": value for (key, value) in audio_checkpoint.items()})
224224

225225
model = _convert_model(original_checkpoint, model, convert_list, device, config)
226+
227+
# TODO: set generation config
226228

227229
model.save_pretrained(pytorch_dump_folder_path)
228230

0 commit comments

Comments
 (0)