a
    d                     @   s.   d dl Z d dlmZ ddlmZ d	ddZdS )
    N)load_file_from_url   )BackboneFcudac                 C   sp   | dkr&t ddddd }d}nt|  dt|d	d
d |d}|jt|d
d |  ||}|S )NZarcface2   g333333?Zir_se)Z
num_layersZ
drop_ratiomoder   z\https://github.com/xinntao/facexlib/releases/download/v0.1.0/recognition_arcface_ir_se50.pthz is not implemented.zfacexlib/weightsT)urlZ	model_dirprogress	file_namesave_dir)strict)r   toevalNotImplementedErrorr   Zload_state_dicttorchload)Z
model_nameZhalfZdeviceZmodel_rootpathmodelZ	model_urlZ
model_path r   f/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/facexlib/recognition/__init__.pyinit_recognition_model   s    

r   )Fr   N)r   Zfacexlib.utilsr   Zarcface_archr   r   r   r   r   r   <module>   s   