Ë
    ¹rœhr  ã                   ó(   — d dl mZ ddlmZ defd„Zy)é    )Ú
DataLoaderé   )Úis_torch_xla_availableÚ
dataloaderc                 óÚ   — t        «       r`dd lmc m} t	        | |j
                  «      sJ d«       ‚dd lmc m} |j                  |j                  «       d«      }|| j                  d<   | S | S )Nr   zPThe dataloader must be a `torch_xla.distributed.parallel_loader.MpDeviceLoader`.)ÚfsdpNÚinput_sharding)r   Ú%torch_xla.distributed.parallel_loaderÚdistributedÚparallel_loaderÚ
isinstanceÚMpDeviceLoaderÚtorch_xla.distributed.spmdÚspmdÚShardingSpecÚget_global_meshÚ_parallel_loader_kwargs)r   ÚplÚxsÚsharding_specs       úp/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/transformers/integrations/tpu.pyÚtpu_spmd_dataloaderr      sl   € ÜÔß:Ð:ä˜* b×&7Ñ&7Ô8ð 	
Ø^ó	
Ð8÷ 	0Ð/àŸ™¨×(:Ñ(:Ó(<¸nÓMˆØ?Lˆ
×*Ñ*Ð+;Ñ<ØÐàÐó    N)Útorch.utils.datar   Úutilsr   r   © r   r   ú<module>r      s   ðõ (å *ð Jô r   