a hV@sJddlZddlZddlZddlmZddlmZmZeGdddZdS)N) dataclass)AnyUnionc@sxeZdZUdZdZeed<eddZe e e j fdddZ ee efd d d Zd d ZddZddZddZdS)DistributedConfigz, Base class for distributed configs Fenable_expert_parallelcKs^|fi|}g}|D](\}}t||rt|||||q|D]}||dqH|S)a Constructs a DistributedConfig instance from a dictionary of parameters. Args: config_dict (Dict[str, Any]): Dictionary containing configuration parameters. **kwargs: Additional keyword arguments to override dictionary values. Returns: DistributedConfig: Instance of DistributedConfig constructed from the dictionary. N)itemshasattrsetattrappendpop)cls config_dictkwargsconfig to_removekeyvaluerh/var/www/html/assistant/venv/lib/python3.9/site-packages/transformers/distributed/configuration_utils.py from_dicts    zDistributedConfig.from_dict)json_file_pathcCsXt|ddd6}|}tj|dddd}||Wdn1sJ0YdS) a 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 `QuantizationConfig()` is serialized to JSON file. wzutf-8)encodingT)indent sort_keys N)opento_dictjsondumpswrite)selfrwriterr Z json_stringrrr to_json_file4s zDistributedConfig.to_json_file)returncCs t|jS)z Serializes this instance to a Python dictionary. Returns: `Dict[str, Any]`: Dictionary of all the attributes that make up this configuration instance. )copydeepcopy__dict__r"rrrrDszDistributedConfig.to_dictccs(t|jD]\}}||fVqdS)zTallows `dict(obj)` for situations where obj may be a dict or QuantizationConfigMixinN)r&r'r(r)r"attrrrrr__iter__LszDistributedConfig.__iter__cCs|jjd|S)N ) __class____name__to_json_stringr)rrr__repr__RszDistributedConfig.__repr__cCstj|jdddS)z Serializes this instance to a JSON formatted string. Returns: str: JSON formatted string representing the configuration instance. r)rr)rr r(r)rrrr/Usz DistributedConfig.to_json_stringc sPg|D](\}}t||r t||||q fdd|D}|S)a Updates attributes of this class instance with attributes from `kwargs` if they match existing attributes, returning all the unused kwargs. Args: kwargs (`Dict[str, Any]`): Dictionary of attributes to tentatively update this class. Returns: `Dict[str, Any]`: Dictionary containing all the key-value pairs that were not used to update the instance. csi|]\}}|vr||qSrr).0rrrrr nz,DistributedConfig.update..)rrr r )r"rrrZ unused_kwargsrr2rupdate]s    zDistributedConfig.updateN)r. __module__ __qualname____doc__rbool__annotations__ classmethodrrstrosPathLiker$dictrrr+r0r/r5rrrrrs   r) r&rr= dataclassesrtypingrrrrrrrs