o h @sBddlZddlZddlZddlZddlmZddlmZddlm Z ddl Z dZ de dee e ffdd Z d d Zd d Ze dfde dede ddfddZd'ddZeddlmZmZddlmZddlmZededeedeededededed zSdd!lmZe je d"d" Z!e d"d" Z"e!j#$%Z&ee!gZ'e"e!Z(e($Z)e)*e'+e!j#$%Z,e&e,ksJed#ed$e-dWdSe.yeed%e/d&ee-dYdSe0y Z1zee1ee-dWYdZ1[1dSdZ1[1ww)(N)warn)Tuple)isdir<command_stringreturncs(ddfdd}||\}}||fS)NcSstdd|DS)Ncss|] }|dVqdS)zUTF-8N)decodestrip).0 to_decoder K/var/www/html/ai/venv/lib/python3.10/site-packages/bitsandbytes/__main__.py s  z6execute_and_return.._decode..)tuple)subprocess_err_out_tupler r r _decodesz#execute_and_return.._decodecs"tjt|tjtjdS)N)stdoutstderr) subprocessPopenshlexsplitPIPE communicate)rrr r &execute_and_return_decoded_std_streamsszBexecute_and_return..execute_and_return_decoded_std_streamsr )rrstd_outstd_errr rr execute_and_returns  rcCs4d|d|}t|\}}t|dkrtd|S)Nzfind z -name rzTSomething when wrong when trying to find file. Maybe you do not have a linux system?)rlen RuntimeError)folderfilenamecmdouterrr r r find_file_recursive!s   r&cCs2tdtdtdtddtjvr)ttjdd}tdt|tdtdr>tdd}tdt|tdttrWttd}td t|tdtd d tjvrtjd }t| d D]$}zt|rt|d t|d}t|Wqntd|YqntddS)NzBUG REPORT INFORMATION CONDA_PREFIXz*cuda*sozANACONDA CUDA PATHSz /usr/local/z /usr/localz/usr/local CUDA PATHSzWORKING DIRECTORY CUDA PATHSzLD_LIBRARY CUDA PATHSLD_LIBRARY_PATH:z CUDA PATHSz Could not read LD_LIBRARY_PATH: ) print_headerprintosenvironr&rgetcwdr setr)pathslib_pathpathr r r generate_bug_report_information*s@      r4+txtwidthfillercCs(|rd|dnd}t|||dS)N r')r,center)r6r7r8r r r r+Rsr+cCstdtddS)Nz] Above we output some debug information. Please provide this info when creating an issue via z/issues/new/choose ... )r,PACKAGE_GITHUB_URLr r r r print_debug_infoYs r<)COMPILED_WITH_CUDAr;) to_be_ignored)get_compute_capabilitiesOTHERzCOMPILED_WITH_CUDA = zCOMPUTE_CAPABILITIES_PER_GPU = r'zDEBUG INFO ENDzY Running a quick check that: + library is importable + CUDA function is callable zK WARNING: Please be sure to sanitize sensible info from any such env vars! )Adam zSUCCESS!zInstallation was successful!z 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!)rN)2r-sysrrwarningsrtypingros.pathrtorch HEADER_WIDTHstrrr&r4intr+r<r'r>r;cuda_setup.env_varsr?cuda_setup.mainr@r,bitsandbytes.optimrBnn Parameterrandcudapadatasumitemp1adamr$lossbackwardstepp2exit ImportError __package__ Exceptioner r r r s    )