Skip to content

Commit 62a02cd

Browse files
authored
deps(conda): use transformers environment with autogptq (#1555)
1 parent 949da77 commit 62a02cd

File tree

4 files changed

+5
-4
lines changed

4 files changed

+5
-4
lines changed

backend/python/autogptq/Makefile

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
.PHONY: autogptq
22
autogptq:
3-
@echo "Creating virtual environment..."
4-
@conda env create --name autogptq --file autogptq.yml
5-
@echo "Virtual environment created."
3+
$(MAKE) -C ../common-env/transformers
4+

backend/python/autogptq/run.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
export PATH=$PATH:/opt/conda/bin
77

88
# Activate conda environment
9-
source activate autogptq
9+
source activate transformers
1010

1111
# get the directory where the bash script is located
1212
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"

backend/python/common-env/transformers/transformers-nvidia.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,7 @@ dependencies:
9393
- tzdata==2023.3
9494
- urllib3==1.26.17
9595
- xxhash==3.4.1
96+
- auto-gptq==0.6.0
9697
- yarl==1.9.2
9798
- soundfile
9899
- langid

backend/python/common-env/transformers/transformers.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@ dependencies:
8080
- triton==2.1.0
8181
- typing-extensions==4.8.0
8282
- tzdata==2023.3
83+
- auto-gptq==0.6.0
8384
- urllib3==1.26.17
8485
- xxhash==3.4.1
8586
- yarl==1.9.2

0 commit comments

Comments
 (0)