o
    iH                     @   s  d dl Z d dlZd dlZd dlmZ d dlmZmZmZ d dl	m
Z
 d dlmZ d dlmZ d dlmZ d dlmZ d d	lmZ d d
lmZ d dlmZ d dlmZ d dlmZ d dlmZmZ d dl m!Z! e
ej"e
dkrvd dl#m$Z$ nedddZ$e%ddG dd deZ&dS )    N)contextmanager)DictOptionalTuple)LooseVersion)tables)postprocess_utils)DatadirWriter)
Transducer)force_gatherable)CTCPrefixScorer)LabelSmoothingLoss)LengthBonus)get_transducer_task_io)load_audio_text_image_videoextract_fbank)BeamSearchTransducerz1.6.0)autocastTc                 c   s    d V  d S )N )enabledr   r   K/home/ubuntu/.local/lib/python3.10/site-packages/funasr/models/bat/model.pyr      s   
r   model_classesBATc                   @   s   e Zd ZdS )r   N)__name__
__module____qualname__r   r   r   r   r   #   s    )T)'timetorchlogging
contextlibr   typingr   r   r   distutils.versionr   funasr.registerr   funasr.utilsr   funasr.utils.datadir_writerr	   funasr.models.transducer.modelr
   funasr.train_utils.device_funcsr   %funasr.models.transformer.scorers.ctcr   "funasr.losses.label_smoothing_lossr   .funasr.models.transformer.scorers.length_bonusr   *funasr.models.transformer.utils.nets_utilsr   funasr.utils.load_utilsr   r   /funasr.models.transducer.beam_search_transducerr   __version__torch.cuda.ampr   registerr   r   r   r   r   <module>   s.   
