diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 682e8051..265e813f 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -14,6 +14,7 @@ dependencies: - python=3.8.* - pytorch-cuda=11.8 - cuda-nvcc=11.8 + - cuda-libraries-dev=11.8 - eventlet=0.33.3 - dnspython=2.2.1 - markdown @@ -57,6 +58,8 @@ dependencies: - scipy - https://github.com/0cc4m/exllama/releases/download/0.0.7/exllama-0.0.7-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' - https://github.com/0cc4m/exllama/releases/download/0.0.7/exllama-0.0.7-cp38-cp38-win_amd64.whl; sys_platform == 'win32' + - https://github.com/henk717/exllamav2/releases/download/0.4/exllamav2-0.0.4-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' + - https://github.com/henk717/exllamav2/releases/download/0.4/exllamav2-0.0.4-cp38-cp38-win_amd64.whl; sys_platform == 'win32' - windows-curses; sys_platform == 'win32' - pynvml - xformers==0.0.21