Fix nested configs of Jukebox (#25533)

This commit is contained in:
Sylvain Gugger 2023-08-16 11:48:24 +02:00 committed by GitHub
parent c385de2441
commit 5ccf343aeb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -606,3 +606,9 @@ class JukeboxConfig(PretrainedConfig):
"""
prior_config_list = [config.to_dict() for config in prior_configs]
return cls(prior_config_list=prior_config_list, vqvae_config_dict=vqvae_config.to_dict(), **kwargs)
def to_dict(self):
# Override the default to_dict to apply to_dict to the list of prior configs.
result = super().to_dict()
result["prior_config_list"] = [config.to_dict() for config in result.pop("prior_configs")]
return result