ݘhVdddlZddlZddlZddlmZddlmZmZeGddZdS)N) dataclass)AnyUnionceZdZUdZdZeed<edZde e e j ffdZ dee effdZd Zd Zd Zd Zd S)DistributedConfigz, Base class for distributed configs Fenable_expert_parallelc |di|}g}|D];\}}t||r&t|||||<|D]}||d|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_removekeyvalues t/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/transformers/distributed/configuration_utils.py from_dictzDistributedConfig.from_dicts##{##  ,,.. & &JCvs## &U+++  %%% " "C JJsD ! ! ! ! json_file_pathct|dd5}|}tj|dddz}||ddddS#1swxYwYdS) 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_strings r to_json_filezDistributedConfig.to_json_file4s.# 8 8 8 &F,,..K*[dKKKdRK LL % % %  & & & & & & & & & & & & & & & & & &sAA$$A(+A(returnc4tj|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's rr#zDistributedConfig.to_dictDs }T]+++rc#xKtj|jD] \}}||fV dS)zTallows `dict(obj)` for situations where obj may be a dict or QuantizationConfigMixinN)r-r.r/r )r'attrrs r__iter__zDistributedConfig.__iter__LsN=77==??  KD%+      rcH|jjd|S)N ) __class____name__to_json_stringr0s r__repr__zDistributedConfig.__repr__Rs'.)CCD,?,?,A,ACCCrc>tj|jddzS)z Serializes this instance to a JSON formatted string. Returns: str: JSON formatted string representing the configuration instance. r)rr!)r$r%r/r0s rr8z DistributedConfig.to_json_stringUs z$-222T99rc g|D];\}}t||r&t||||<fd|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. c$i|] \}}|v || Sr r ).0rrrs r z,DistributedConfig.update..ns*]]] US\H\H\eH\H\H\r)r r r r)r'rrr unused_kwargsrs @rupdatezDistributedConfig.update]s  ,,.. & &JCtS!! &c5)))  %%%^]]]fllnn]]] rN)r7 __module__ __qualname____doc__rbool__annotations__ classmethodrrstrosPathLiker*dictrr#r3r9r8r@r rrrrs$)D((([(&5bk1A+B&&&& ,c3h,,,, DDD:::rr) r-r$rH dataclassesrtypingrrrr rrrMs !!!!!! XXXXXXX XXXr