simcop2387/AutoGPTQ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

disable the error exit here, see if the pregen code works

Browse source
This commit is contained in:
Ryan Voots 2023-10-26 12:43:07 -04:00
parent 07021b9a1c
commit ced04e1dff
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
setup.py
View file

@ -99,9 +99,9 @@ if BUILD_CUDA_EXT:
if platform.system() != 'Windows':
p = int(subprocess.run("cat /proc/cpuinfo | grep cores | head -1", shell=True, check=True, text=True, stdout=subprocess.PIPE).stdout.split(" ")[2])
ret = subprocess.call(["python", "./autogptq_extension/qigen/generate.py", "--module", "--search", "--p", str(p)])
if ret != 0:
raise Exception(f"Failed generate with {ret}")
sys.exit(-1)
# if ret != 0:
# raise Exception(f"Failed generate with {ret}")
# sys.exit(-1)
if not ROCM_VERSION:
from distutils.sysconfig import get_python_lib