gbddlZddlZddlZddlmZGddejZd dedefdZdS) N) PartialStatec`eZdZdZedZdZejddZ dS)MultiProcessAdaptera@ An adapter to assist with logging in multiprocess. `log` takes in an additional `main_process_only` kwarg, which dictates whether it should be called on all processes or only the main executed one. Default is `main_process_only=True`. Does not require an `Accelerator` object to be created first. c6t}| p|o|jS)z Check if log should be performed)ris_main_process)main_process_onlystates N/var/www/html/ai-engine/env/lib/python3.11/site-packages/accelerate/logging.py _should_logzMultiProcessAdapter._should_log s$$$U):)Tu?TUcttjikrtd|dd}|dd}|dd||r||r2|||\}}|jj ||g|Ri|dS|rvt}t|j D]U}||j kr0|||\}}|jj ||g|Ri|| RdSdSdS) a\ Delegates logger call after checking if we should log. Accepts a new kwarg of `main_process_only`, which will dictate whether it will be logged across all processes or only the main executed one. Default is `True` if not passed Also accepts "in_order", which if `True` makes the processes log one by one, in order. This is much easier to read, but comes at the cost of sometimes needing to wait for the other processes. Default is `False` to not break with the previous behavior. `in_order` is ignored if `main_process_only` is passed. zYou must initialize the accelerate state by calling either `PartialState()` or `Accelerator()` before using the logging utility.r Tin_orderF stacklevelN)r _shared_state RuntimeErrorpop setdefault isEnabledForr processloggerlogrange num_processes process_indexwait_for_everyone) selflevelmsgargskwargsr rr is r rzMultiProcessAdapter.log&s  % + +S #JJ':DAA::j%00,***   U # # . 122 ."ll377 V s>> from accelerate.logging import get_logger >>> from accelerate import Accelerator >>> logger = get_logger(__name__) >>> accelerator = Accelerator() >>> logger.info("My log", main_process_only=False) >>> logger.debug("My log", main_process_only=True) >>> logger = get_logger(__name__, log_level="DEBUG") >>> logger.info("My log") >>> logger.debug("My second log") >>> array = ["a", "b", "c", "d"] >>> letter_at_rank = array[accelerator.process_index] >>> logger.info(letter_at_rank, in_order=True) ``` NACCELERATE_LOG_LEVEL) osenvirongetlogging getLoggersetLevelupperrootr)r/r0rs r get_loggerr;UsDJNN#94@@  t $ $F ))*** Y__../// vr * **r )N) r,r6r3r r LoggerAdapterrstrr;r.r r r>s <&<&<&<&<&'/<&<&<&~(+(+S(+S(+(+(+(+(+(+r