From 110f8229c565a1ac64060e4e1785d4563920d4f4 Mon Sep 17 00:00:00 2001 From: 0cc4m Date: Sat, 1 Apr 2023 21:33:05 +0200 Subject: [PATCH] Add cudatoolkit-dev for compilation, compatible gcc 9 and update transformers to fix error in gptq --- environments/huggingface.yml | 5 ++++- environments/rocm.yml | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 6807627e..71d26e9c 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -11,6 +11,9 @@ dependencies: - pytorch=1.11.* - python=3.8.* - cudatoolkit=11.1 + - cudatoolkit-dev=11.1 + - gcc=9.* + - gxx=9.* - eventlet=0.33.3 - dnspython=2.2.1 - markdown @@ -30,7 +33,7 @@ dependencies: - flask-ngrok - flask-cors - lupa==1.10 - - git+https://github.com/huggingface/transformers@88dae78f4d204428568f749e864ef5ba09da7d24 + - git+https://github.com/huggingface/transformers@c612628045822f909020f7eb6784c79700813eda - git+https://github.com/huggingface/datasets@test-pandas-2.0.0rc - huggingface_hub==0.12.1 - safetensors diff --git a/environments/rocm.yml b/environments/rocm.yml index a1d3d8b0..dda2a2b2 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -29,7 +29,7 @@ dependencies: - flask-ngrok - flask-cors - lupa==1.10 - - git+https://github.com/huggingface/transformers@88dae78f4d204428568f749e864ef5ba09da7d24 + - git+https://github.com/huggingface/transformers@c612628045822f909020f7eb6784c79700813eda - huggingface_hub==0.12.1 - safetensors - accelerate