o ¶ñyfîã@s&ddlmZGdd„dƒZdd„ZdS)é)Úsharedc@seZdZdd„Zdd„ZdS)ÚFaceRestorationcCsdS)NÚNone©)ÚselfrrúiC:\Users\Luke\Documents\Stable diffusion\Automatic1111\stable-diffusion-webui\modules\face_restoration.pyÚnameózFaceRestoration.namecCs|S©Nr)rÚnp_imagerrrÚrestorer zFaceRestoration.restoreN)Ú__name__Ú __module__Ú __qualname__rr rrrrrs rcCs2dd„tjDƒ}t|ƒdkr|S|d}| |¡S)NcSs,g|]}| ¡tjjkstjjdur|‘qSr )rrÚoptsÚface_restoration_model)Ú.0ÚxrrrÚ s,z!restore_faces..r)rÚface_restorersÚlenr )r rZ face_restorerrrrÚ restore_faces s   rN)ÚmodulesrrrrrrrÚs