
    Pip                     b    d dl mZmZ d dlmZ d dlZ	  G d de          Z G d d          ZdS )    )ABCabstractmethod)AnyNc            	       ^    e Zd Zedej        j        dededej        j        fd            ZdS )	Quantizermodelargskwargsreturnc                     d S N selfr   r	   r
   s       p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchao/quantization/unified.pyquantizezQuantizer.quantize   	     	    N)	__name__
__module____qualname__r   torchnnModuler   r   r   r   r   r   r      sZ        X_-0<?	   ^  r   r   c            	           e Zd Zedej        j        dededej        j        fd            Zedej        j        dededej        j        fd            Z	dS )TwoStepQuantizerr   r	   r
   r   c                     d S r   r   r   s       r   preparezTwoStepQuantizer.prepare   r   r   c                     d S r   r   r   s       r   convertzTwoStepQuantizer.convert%   r   r   N)
r   r   r   r   r   r   r   r   r   r    r   r   r   r   r      s        X_-0<?	   ^
 X_-0<?	   ^  r   r   )abcr   r   typingr   r   r   r   r   r   r   <module>r#      s    $ # # # # # # #                       r   