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

fix install bug

Browse source
This commit is contained in:
qwopqwop200 2023-08-31 15:00:38 +09:00 • committed by GitHub
parent 45a1ee4d84
commit f97b77a64e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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

@ -95,8 +95,8 @@ include_dirs = ["autogptq_cuda"]
additional_setup_kwargs = dict() additional_setup_kwargs = dict()
if BUILD_CUDA_EXT: if BUILD_CUDA_EXT:
from torch.utils import cpp_extension from torch.utils import cpp_extension
if platform != 'Windows': 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]) p = int(subprocess.run("cat /proc/cpuinfo | grep cores | head -1", shell=True, check=True, text=True, stdout=subprocess.PIPE).stdout.split(" ")[2])
subprocess.call(["python", "./autogptq_extension/qigen/generate.py", "--module", "--search", "--p", str(p)]) subprocess.call(["python", "./autogptq_extension/qigen/generate.py", "--module", "--search", "--p", str(p)])
@ -125,7 +125,7 @@ if BUILD_CUDA_EXT:
) )
] ]
if platform != 'Windows': if platform.system() != 'Windows':
extensions.append( extensions.append(
cpp_extension.CppExtension( cpp_extension.CppExtension(
"cQIGen", "cQIGen",