Merge branch 'master' into resumable_http

This commit is contained in:
Thomas Wolf
2019-11-27 17:10:36 +01:00
committed by GitHub
131 changed files with 11504 additions and 921 deletions

View File

@@ -57,6 +57,7 @@ class PretrainedConfig(object):
self.torchscript = kwargs.pop('torchscript', False) # Only used by PyTorch models
self.use_bfloat16 = kwargs.pop('use_bfloat16', False)
self.pruned_heads = kwargs.pop('pruned_heads', {})
self.is_decoder = kwargs.pop('is_decoder', False)
def save_pretrained(self, save_directory):
""" Save a configuration object to the directory `save_directory`, so that it