o ϯi@sddlmZmZedZedZdZddZddlmZm Z edZe dZdZdd Z dd lm Z m Z e d Ze d ZdZd d Z dS)) BertTokenizer BertModelzbert-base-uncasedz"Replace me by any text you'd like.cCt|dd}tdi|}|SNpt)return_tensors tokenizermodeltext encoded_inputoutputrrO/home/ubuntu/.local/lib/python3.10/site-packages/laion_clap/clap_module/bert.pybert_embeddings r)RobertaTokenizer RobertaModelz roberta-basecCrrr r rrrRoberta_embeddingsrr) BartTokenizer BartModelzfacebook/bart-basecCrrr r rrrbart_embeddingsrrN) transformersrrfrom_pretrainedr r r rrrrrrrrrrrs