a
    þdž  ã                   @   s$  d dl mZ d dlmZmZ d dlmZmZmZm	Z	m
Z
 d dlmZmZmZmZ d dlmZmZmZ d dlmZmZmZmZmZmZmZmZmZ d dlmZmZ d dl m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z( d d	l)m*Z*m)Z) d d
l+m,Z,m-Z-m.Z. d dl/m0Z0m1Z1 d dl2m3Z3m4Z4 d dl5m6Z6m7Z7m8Z8 dS )é   )ÚCoCa)ÚOPENAI_DATASET_MEANÚOPENAI_DATASET_STD)Úcreate_modelÚcreate_model_and_transformsÚcreate_model_from_pretrainedÚget_tokenizerÚcreate_loss)Úlist_modelsÚadd_model_configÚget_model_configÚload_checkpoint)ÚClipLossÚDistillClipLossÚCoCaLoss)	ÚCLIPÚCustomTextCLIPÚCLIPTextCfgÚCLIPVisionCfgÚconvert_weights_to_lpÚconvert_weights_to_fp16Útrace_modelÚget_cast_dtypeÚget_input_dtype)Úload_openai_modelÚlist_openai_models)Úlist_pretrainedÚlist_pretrained_models_by_tagÚlist_pretrained_tags_by_modelÚget_pretrained_urlÚdownload_pretrained_from_urlÚis_pretrained_cfgÚget_pretrained_cfgÚdownload_pretrained)Úpush_pretrained_to_hf_hubÚpush_to_hf_hub)ÚSimpleTokenizerÚtokenizeÚdecode)Úimage_transformÚAugmentationCfg)Úbuild_zero_shot_classifierÚ!build_zero_shot_classifier_legacy)ÚOPENAI_IMAGENET_TEMPLATESÚSIMPLE_IMAGENET_TEMPLATESÚIMAGENET_CLASSNAMESN)9Z
coca_modelr   Ú	constantsr   r   Úfactoryr   r   r   r   r	   r
   r   r   r   Zlossr   r   r   Úmodelr   r   r   r   r   r   r   r   r   Zopenair   r   Z
pretrainedr   r   r   r   r    r!   r"   r#   r%   r$   Ú	tokenizerr&   r'   r(   Z	transformr)   r*   Zzero_shot_classifierr+   r,   Zzero_shot_metadatar-   r.   r/   © r4   r4   ú[/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/open_clip/__init__.pyÚ<module>   s   ,(