o
    *Îj¹  ã                   @   sB   d dl Z d dlZd dlmZ dd„ Zdd„ Zedkreƒ  dS dS )é    N)Úinit_megatron_utilc                  C   s$   dddddœ} dt jd< t| ƒ d S )Né   Zncclé*   )Ztensor_model_parallel_sizeZ
world_sizeZdistributed_backendÚseedZ39501ZMASTER_PORT)ÚosÚenvironr   )Zdummy_megatron_cfg© r   ú]/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/utils/pre_compile.pyÚpre_compile_megatron_util   s   ü
r
   c                  C   s$   t j ¡ rddlm}  tƒ  d S d S )Nr   ©Úms_deform_attn_func)ÚtorchÚcudaZis_availableZ0easycv.thirdparty.deformable_attention.functionsr   r
   r   r   r   r	   Úpre_compile_all   s   

ür   Ú__main__)r   r   Zmodelscope.utils.megatron_utilsr   r
   r   Ú__name__r   r   r   r	   Ú<module>   s    
ÿ