Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 18:45:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Don't assume that py-cpuinfo is installed

Browse source
This commit is contained in:
oobabooga 2023-09-24 08:10:45 -07:00
parent eb0b7c1053
commit d5952cb540
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
one_click.py
View file

@ -40,14 +40,17 @@ def is_x86_64():
def cpu_has_avx2(): def cpu_has_avx2():
import cpuinfo try:
import cpuinfo
info = cpuinfo.get_cpu_info() info = cpuinfo.get_cpu_info()
if 'avx2' in info['flags']: if 'avx2' in info['flags']:
return True
else:
return False
except:
return True return True
return False
def torch_version(): def torch_version():
from torch import __version__ as torver from torch import __version__ as torver
@ -157,7 +160,7 @@ def install_webui():
install_pytorch = "python -m pip install torch==2.0.1a0 torchvision==0.15.2a0 intel_extension_for_pytorch==2.0.110+xpu -f https://developer.intel.com/ipex-whl-stable-xpu" install_pytorch = "python -m pip install torch==2.0.1a0 torchvision==0.15.2a0 intel_extension_for_pytorch==2.0.110+xpu -f https://developer.intel.com/ipex-whl-stable-xpu"
# Install Git and then Pytorch # Install Git and then Pytorch
run_cmd(f"{install_git} && {install_pytorch} && python -m pip install py-cpuinfo", assert_success=True, environment=True) run_cmd(f"{install_git} && {install_pytorch} && python -m pip install py-cpuinfo==9.0.0", assert_success=True, environment=True)
# Install the webui requirements # Install the webui requirements
update_requirements(initial_installation=True) update_requirements(initial_installation=True)