a
    dR                     @   sB   d dl Z d dlmZ ddlmZ ddlmZ ddgZdd
dZdS )    N)load_file_from_url   )FAN)landmark_98_to_68r   r   Fcudac                 C   sj   | dkrt dd|d}d}nt|  dt|ddd |d	}|jt|d
 dd |  ||}|S )NZ	awing_fan   b   )Znum_modulesZnum_landmarksdevicezShttps://github.com/xinntao/facexlib/releases/download/v0.1.0/alignment_WFLW_4HG.pthz is not implemented.zfacexlib/weightsT)urlZ	model_dirprogress	file_namesave_dirZ
state_dict)strict)r   NotImplementedErrorr   Zload_state_dicttorchloadevalto)Z
model_nameZhalfr	   Zmodel_rootpathmodelZ	model_urlZ
model_path r   d/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/facexlib/alignment/__init__.pyinit_alignment_model
   s    

r   )Fr   N)	r   Zfacexlib.utilsr   Z
awing_archr   Zconvert_98_to_68_landmarksr   __all__r   r   r   r   r   <module>   s
   