o
    ϯi#
                     @   s  d Z ddlZddlZddlZddlZdd Zdd ZejddZ	e	
  d	d
gZe	jeddZeeddddf  eej ejd	dd\ZZeddZe	jeddZeeddddf  eej d	d
gZe	jeddZeeddddf  eej ejd	dd\ZZeddZeeee Ze	jeddZeeddddf  eej ddgZe	eZee eej ddgZe	jeddZee eej dS )z
Contrastive Language-Audio Pretraining Model from LAION
--------------------------------------------------------
Paper: https://arxiv.org/abs/2211.06687
Authors (equal contributions): Ke Chen, Yusong Wu, Tianyu Zhang, Yuchen Hui
Support: LAION
    Nc                 C   s   | d  dS )N    @float32)astypex r   H/home/ubuntu/.local/lib/python3.10/site-packages/laion_clap/unit_test.pyint16_to_float32   s   r	   c                 C   s   t j| ddd} | d dS )Ng      g      ?)a_mina_maxr   int16)npclipr   r   r   r   r   float32_to_int16   s   r   F)enable_fusionz9/home/la/kechen/Research/KE_CLAP/ckpt/test_clap_short.wavz8/home/la/kechen/Research/KE_CLAP/ckpt/test_clap_long.wav)r   
use_tensorii  )sr   TzI love the contrastive learningzI love the pretrain model)r   )__doc__numpyr   librosatorch
laion_clapr	   r   CLAP_Modulemodel	load_ckpt
audio_file!get_audio_embedding_from_filelistaudio_embedprintshapeload
audio_data_reshapeget_audio_embedding_from_data
from_numpyfloat	text_dataget_text_embedding
text_embedr   r   r   r   <module>   sP    





