
    ;i                         S SK r S SKrS SKrS SKrS SKrS SKJr  S SKJ	r	  S SK
Jr  S SKJr  S SKJr  S SKJrJr  / SQrS	 rS
\S\4S jrS rS rg)    N)__version__)BNB_BACKEND)PACKAGE_GITHUB_URL)get_cuda_specs)print_diagnostics)print_dedentedprint_header)	
accelerate	diffusersnumpypippeftsafetensorstransformerstritontrlc                     SSK Jn   [        R                  R	                  [        R
                  " SS5      R                  5       5      n[        R
                  " SS5      R                  5       nUR                  R                  5       R                  5       nU " U/5      nX!-  nUR                  5       nUR                  5         UR                  5         UR                  R                  5       R                  5       nX7:w  d   eg )Nr   )Adam
   )bitsandbytes.optimr   torchnn	Parameterrandcudadatasumitembackwardstep)r   pap1adamoutlossp2s           m/home/dmtnaga/Documents/work/airagagent/rag_env/lib/python3.13/site-packages/bitsandbytes/diagnostics/main.pysanity_checkr)      s    '5::b"-2245A

2r!A	
			B9D
%C779DMMOIIK	
			B8O8    namereturnc                      [         R                  R                  U 5      nU$ ! [         R                  R                   a    Sn U$ f = f)Nz	not found)	importlibmetadataversionPackageNotFoundError)r+   r0   s     r(   get_package_versionr2   -   sK    $$,,T2 N 22 Ns   #  AAc                     [        S[        R                  " 5        35        [        R                  " 5       S:X  a0  [        SSR                  [        R                  " 5       5       35        [        S[        R
                  " 5        35        [        S[        R                   35        [        S[        R                  R                  =(       d    S 35        [        S	[        R                  R                  =(       d    S 35        [        S
[        [        R                  SS5      =(       d    S 35        [        S5        [         H  n [        U 5      n[        SU  SU 35        M!     g)z4Simple utility to print out environment information.z
Platform: Linuxz  libc: -zPython: z	PyTorch: z  CUDA: zN/Az  HIP: z  XPU: xpuzRelated packages:z  z: N)printplatformsystemjoinlibc_verpython_versionr   r   r0   r   hipgetattr_RELATED_PACKAGESr2   )pkgr0   s     r(   show_environmentrA   5   s    
Jx((*+
,-G#("3"3"56789	HX,,./
01	Ie''(
)*	HU]]''051
23	GEMM%%./
01	GGEMM5%8AEB
CD	
 %c*3%r'#$ !r*   c                     [        S[         35        [        5         [        S5        [        5       n U (       a  [	        U 5        [
        R                  R                  5       (       d=  [        S[         S35        [        S[         S35        [        S5        [        S5        g [        S	[         S
35         [        5         [        S5        g ! [         a.  nS[        U5      ;   a  [        S[         S35         S nAO*UeS nAf[         a    [        R                   " 5          Of = f[#        S[$         S35        [&        R(                  " S5        g )Nzbitsandbytes v zPyTorch says z$ is not available. Possible reasons:z1. z driver not installedz!2. Using a CPU-only PyTorch buildz3. No GPU detectedz,Checking that the library is importable and z is callable...zSUCCESS!znot available in CPU-onlyz	WARNING: z is currently running as CPU-only!
Therefore, 8-bit optimizers and GPU quantization are unavailable.

If you think that this is so erroneously,
please report an issue!zu
            Above we output some debug information.
            Please provide this info when creating an issue via z/issues/new/choose
            WARNING: Please be sure to sanitize sensitive info from the output before posting it.
               )r	   bnb_versionrA   r   r   r   r   is_availabler7   r   r)   RuntimeErrorstr__package__	Exception	traceback	print_excr   r   sysexit)
cuda_specses     r(   mainrQ   I   s+   >+/0!J*%
 ::""$$k]*NOPK= 56712"# 	<[MYZ	"N* 	*c!f4} -Y Z   	"!	" 	AAS@T U	
 	s$   8C 
D&"D?D"D&%D&)r.   r8   rM   rK   r   bitsandbytesr   rE   bitsandbytes.cextensionr   bitsandbytes.constsr   bitsandbytes.cuda_specsr   bitsandbytes.diagnostics.cudar   bitsandbytes.diagnostics.utilsr   r	   r?   r)   rH   r2   rA   rQ    r*   r(   <module>rY      sU      
   3 / 2 2 H
 c c %(-r*   