o
    ci                     @   s0   d dl Z d dlZG dd dZe jdd ZdS )    Nc                   @   s$   e Zd Zdd Zdd Zdd ZdS )_FilteredStdoutc                 C   s   || _ || _d S N)_stream	_suppress)selfstreamsuppress_substrings r	   L/home/ubuntu/vllm_env/lib/python3.10/site-packages/faster_qwen3_tts/utils.py__init__   s   
z_FilteredStdout.__init__c                    s,   t  fdd| jD rt S | j S )Nc                 3   s    | ]}| v V  qd S r   r	   ).0sdatar	   r
   	<genexpr>   s    z(_FilteredStdout.write.<locals>.<genexpr>)anyr   lenr   write)r   r   r	   r   r
   r   
   s   z_FilteredStdout.writec                 C   s
   | j  S r   )r   flush)r   r	   r	   r
   r      s   
z_FilteredStdout.flushN)__name__
__module____qualname__r   r   r   r	   r	   r	   r
   r      s    r   c                  c   sF    t tjdd} t|  d V  W d    d S 1 sw   Y  d S )N)zflash-attn is not installedzmanual PyTorch versionzPlease install flash-attn)r   )r   sysstdout
contextlibredirect_stdout)filteredr	   r	   r
   suppress_flash_attn_warning   s   "r   )r   r   r   contextmanagerr   r	   r	   r	   r
   <module>   s
    