o shV@sJddlZddlZddlZddlmZddlmZmZeGdddZdS)N) dataclass)AnyUnionc@sxeZdZUdZdZeed<eddZde e e j ffddZ d ee effd d Zd d ZddZddZddZdS)DistributedConfigz, Base class for distributed configs Fenable_expert_parallelcKs^|di|}g}|D]\}}t||r!t|||||q |D]}||dq$|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_removekeyvaluerrj/var/www/html/alpaca_bot/venv/lib/python3.10/site-packages/transformers/distributed/configuration_utils.py from_dicts    zDistributedConfig.from_dictjson_file_pathcCsZt|ddd}|}tj|dddd}||WddS1s&wYdS) 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 json_stringrrr to_json_file4s   "zDistributedConfig.to_json_filereturncCs 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"rrrrDs zDistributedConfig.to_dictccs*t|jD] \}}||fVq dS)zTallows `dict(obj)` for situations where obj may be a dict or QuantizationConfigMixinN)r'r(r)r)r"attrrrrr__iter__Ls zDistributedConfig.__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*rrrr0Usz DistributedConfig.to_json_stringc sPg|D]\}}t||rt||||qfdd|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 nsz,DistributedConfig.update..)rr r r )r"rrr unused_kwargsrr3rupdate]s    zDistributedConfig.updateN)r/ __module__ __qualname____doc__rbool__annotations__ classmethodrrstrosPathLiker%dictrrr,r1r0r6rrrrrs    r) r'rr> dataclassesrtypingrrrrrrrs