Bark: flexible generation config overload (#25414)

This commit is contained in:
Joao Gante 2023-08-09 18:51:51 +01:00 committed by GitHub
parent 944ddce8bf
commit d0c1aebea4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -224,7 +224,7 @@ class BarkFineGenerationConfig(GenerationConfig):
self.max_fine_input_length = max_fine_input_length
self.n_fine_codebooks = n_fine_codebooks
def validate(self):
def validate(self, **kwargs):
"""
Overrides GenerationConfig.validate because BarkFineGenerationConfig don't use any parameters outside
temperature.