+
    œ°³j|  ã                  ó    € R t ^ RIHt ^ RIt^RIHtHtHtHtH	t	H
t
HtHtHtHtHtHtHtHtHtHtHtHtHtHtHt ^RIHt ]t]t]
tR t. ROt R# )zŽBackward-compatibility shim for the tensor parallel API.

The canonical implementation lives in ``transformers.distributed.tensor_parallel``.
)ÚannotationsN)ÚALL_PARALLEL_STYLESÚAllReduceParallelÚColwiseParallelÚEpRouterParallelÚMlaKvAProjParallelÚMoeExpertsParallelÚMoeIdentityParallelÚMoEParamShardÚMoeTensorParalellMegaMoeExpertsÚPackedColwiseParallelÚPackedRowwiseParallelÚParallelInterfaceÚReplicatedWithGradAllReduceÚRouterParallelMegaMoeÚRowwiseParallelÚSequenceParallelÚTensorParallelLayerÚapply_tensor_parallelismÚgather_state_dict_for_saveÚ replace_layer_number_by_wildcardÚverify_tp_plan)Úinitialize_tensor_parallelismc                 óR   € \         P                  ! R\        ^R7       \        R4      h)zIDeprecated per-parameter sharding helper from the legacy TP loading path.zä`shard_and_distribute_module` is deprecated and unavailable with the DTensor tensor-parallel loading path. Use `transformers.distributed.tensor_parallel.apply_tensor_parallelism` with `from_pretrained(..., tp_plan=...)` instead.)Ú
stacklevelz[`shard_and_distribute_module` is unavailable with the DTensor tensor-parallel loading path.)ÚwarningsÚwarnÚFutureWarningÚRuntimeError)ÚargsÚkwargss   *,Új/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/integrations/tensor_parallel.pyÚshard_and_distribute_moduler"   6   s)   € ä‡M‚Mð	7ô 	Øõô ÐtÓ
uÐuó    )r   ÚMoeIdentityExpertParallelÚMoeTensorParalellExpertsÚRouterParallelr   r   r   r   r   r   r
   r	   r   r   r   r   r   r   r   r   r   r   r   r   r"   r   )!Ú__doc__Ú
__future__r   r   Údistributed.tensor_parallelr   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   Údistributed.utilsr   r&   r$   r%   r"   Ú__all__© r#   r!   Ú<module>r-      sU   ðñõ
 #ã ÷÷ ÷ ÷ ÷ õ õ. >ð "€Ø/Ð Ø-Ð ò	vò‚r#   