a
    dN                     @   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   )HyperIQAFcudac              	   C   st   | dkr$t dddddddd}d	}nt|  d
t|ddd |d}|jtj|dd d | }||}|S )Nhypernet   p      8            zThttps://github.com/xinntao/facexlib/releases/download/v0.2.0/assessment_hyperIQA.pthz is not implemented.zfacexlib/weightsT)urlZ	model_dirprogress	file_namesave_dirc                 S   s   | S )N )Zstoragelocr   r   e/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/facexlib/assessment/__init__.py<lambda>       z'init_assessment_model.<locals>.<lambda>)Zmap_location)	r   NotImplementedErrorr   r   Zload_state_dicttorchloadevalto)Z
model_nameZhalfZdeviceZmodel_rootpathmodelZ	model_urlZhypernet_model_pathr   r   r   init_assessment_model   s    

r   )Fr   N)r   Zfacexlib.utilsr   Zhyperiqa_netr   r   r   r   r   r   <module>   s   