How to create a config.json after saving a model

BramVanroy · April 28, 2022, 3:35pm

You should be able to just call

model.config.to_json_file("config.json")

github.com

huggingface/transformers/blob/1be8d56ec6f7113810adc716255d371e78e8a1af/src/transformers/configuration_utils.py#L808

      
        
            
            
    Returns:
                    `str`: String containing all the attributes that make up this configuration instance in JSON format.
                """
                if use_diff is True:
                    config_dict = self.to_diff_dict()
                else:
                    config_dict = self.to_dict()
                return json.dumps(config_dict, indent=2, sort_keys=True) + "\n"
            
            
def to_json_file(self, json_file_path: Union[str, os.PathLike], use_diff: bool = True):
                """
                Save this instance to a JSON file.
            
            
    Args:
                    json_file_path (`str` or `os.PathLike`):
                        Path to the JSON file in which this configuration instance's parameters will be saved.
                    use_diff (`bool`, *optional*, defaults to `True`):
                        If set to `True`, only the difference between the config instance and the default `PretrainedConfig()`
                        is serialized to JSON file.
                """

cc @seanbenhur

Topic		Replies	Views
Convert Pytorch Model to Huggingface Transformer? 🤗Transformers	2	11014	May 2, 2024
How to get config.json from .pt loaded model? Beginners	1	1350	April 11, 2022
Save custom transformer as PreTrainedModel Intermediate	1	934	September 7, 2021
How to save custom model to get config.json file Beginners	2	1489	October 20, 2024
Saving trained number of epochs in config.json // Custom fields in config.json 🤗Transformers	1	278	November 7, 2022

How to create a config.json after saving a model

Related topics