Download the cuda branch directly

This commit is contained in:
oobabooga 2023-03-30 18:22:48 -03:00 committed by GitHub
parent 78c0da4a18
commit 85e4ec6e6b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -91,9 +91,8 @@ if not exist repositories\ (
) )
cd repositories || goto end cd repositories || goto end
if not exist GPTQ-for-LLaMa\ ( if not exist GPTQ-for-LLaMa\ (
git clone https://github.com/qwopqwop200/GPTQ-for-LLaMa.git git clone https://github.com/qwopqwop200/GPTQ-for-LLaMa.git -b cuda
cd GPTQ-for-LLaMa || goto end cd GPTQ-for-LLaMa || goto end
git checkout cuda
call python -m pip install -r requirements.txt call python -m pip install -r requirements.txt
call python setup_cuda.py install call python setup_cuda.py install
if not exist "%INSTALL_ENV_DIR%\lib\site-packages\quant_cuda-0.0.0-py3.10-win-amd64.egg" ( if not exist "%INSTALL_ENV_DIR%\lib\site-packages\quant_cuda-0.0.0-py3.10-win-amd64.egg" (