NjR <ddlmZddlZddlmZddlZejeZ Gdde eZ e jeje jeje j ej"e j$ej&iZddZ d dZy) ) annotationsN)EnumceZdZdZdZdZdZy)DTypefloat16float32float64int8N)__name__ __module__ __qualname__Float16Float32Float64Int8g/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/model2vec/quantization.pyrr sGGG Drrc`t|}|j|k(r|S|tjtjtj hvr|j |S|tjk(rtjtj|dz }|j tjd}tj|||tj||tj|dd||j tj}|St!d)a+Quantize embeddings to a specified data type to reduce memory usage. :param embeddings: The embeddings to quantize, as a numpy array. :param quantize_to: The data type to quantize to. :return: The quantized embeddings. :raises ValueError: If the quantization type is not valid. g_@T)copy)outiz!Not a valid enum member of DType.) dtype_mapdtyperrrrastypernpmaxabsrdividerintclipr ValueError) embeddings quantize_to mapped_dtypescalebuf quantizeds rquantize_embeddingsr)s[)L<'u}}emmU]]CC  ..  "rvvj)*U2 6 #u#&  T3C(JJrww' <==rc|t|}t||}|;||jdkDrtd|d|jd|ddd|f}|S)aQuantize embeddings to a datatype and reduce dimensionality. :param embeddings: The embeddings to quantize and reduce, as a numpy array. :param quantize_to: The data type to quantize to. If None, no quantization is performed. :param dimensionality: The number of dimensions to keep. If None, no dimensionality reduction is performed. :return: The quantized and reduced embeddings. :raises ValueError: If the passed dimensionality is not None and greater than the model dimensionality. NzDimensionality z* is greater than the model dimensionality )rr)shaper")r#r$dimensionalitys rquantize_and_reduce_dimr.=sK( ([A ! J,,Q/ /!.!11[\f\l\lmn\o[pq  ?N? 23 r)r# np.ndarrayr$rreturnr/)r#r/r$zstr | DType | Noner-z int | Noner0r/) __future__rloggingenumrnumpyr getLoggerr loggerstrrrrrrrr rr rr)r.rrrr8s"   8 $C MM2:: MM2:: MM2:: JJ    >F);MWr