o
    úÜÓh  ã                   @   s®   d Z ddlmZ ddlmZmZmZmZmZm	Z	m
Z
 ddlmZ ddlmZmZ ddlmZ ddlmZmZmZmZmZmZmZmZmZmZmZmZ eƒ rUddlmZ d	S d	S )
z0.25.0é   )ÚAccelerator)Úcpu_offloadÚcpu_offload_with_hookÚdisk_offloadÚdispatch_modelÚinit_empty_weightsÚinit_on_deviceÚload_checkpoint_and_dispatch)Úskip_first_batches)Údebug_launcherÚnotebook_launcher)ÚPartialState)ÚAutocastKwargsÚDeepSpeedPluginÚDistributedDataParallelKwargsÚDistributedTypeÚFullyShardedDataParallelPluginÚGradScalerKwargsÚInitProcessGroupKwargsÚfind_executable_batch_sizeÚinfer_auto_device_mapÚis_rich_availableÚload_checkpoint_in_modelÚsynchronize_rng_states)ÚrichN) Ú__version__Úacceleratorr   Úbig_modelingr   r   r   r   r   r   r	   Údata_loaderr
   Ú	launchersr   r   Ústater   Úutilsr   r   r   r   r   r   r   r   r   r   r   r   r   © r"   r"   úI/var/www/html/ai/venv/lib/python3.10/site-packages/accelerate/__init__.pyÚ<module>   s    $	8ÿ