
    uj                     H    d dl mZ  G d d          Z G d de          ZdS )    )tqdmc                   0    e Zd ZdedefdZdefdZd ZdS )ProgressCallbackfilename	file_sizec                 "    || _         || _        d S N)r   r   )selfr   r   s      g/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/hub/callback.py__init__zProgressCallback.__init__   s     "    sizec                     d S r	    r
   r   s     r   updatezProgressCallback.update
       r   c                     d S r	   r   r
   s    r   endzProgressCallback.end   r   r   N)__name__
__module____qualname__strintr   r   r   r   r   r   r   r      s`        # # # # # #3        r   r   c                   :     e Zd Zdedef fdZdefdZd Z xZS )TqdmCallbackr   r   c           	          t                                          ||           t          ddd|dk    r|nddd| j        z   dz   d          | _        d S )	NBTi   r      zDownloading [])unit
unit_scaleunit_divisortotalinitialdescleave)superr   r   r   progress)r
   r   r   	__class__s      r   r   zTqdmCallback.__init__   sd    9---(1}}))! 4=036  r   r   c                 :    | j                             |           d S r	   )r*   r   r   s     r   r   zTqdmCallback.update   s    T"""""r   c                 8    | j                                          d S r	   )r*   closer   s    r   r   zTqdmCallback.end!   s    r   )	r   r   r   r   r   r   r   r   __classcell__)r+   s   @r   r   r      su        	 	 	 	 	 	 	 	#3 # # # #      r   r   N)	tqdm.autor   r   r   r   r   r   <module>r1      st         
 
 
 
 
 
 
 
    #     r   