mathieu/ai-suite-rocm-local
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

remove gfx1102 for llama build

Browse Source
This commit is contained in:
Mathieu Broillet 2024-08-28 15:44:59 +02:00
parent 9ba38e0a71
commit be6be8a529
Signed by: mathieu
GPG Key ID: A08E484FE95074C1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prebuilts/Dockerfile
View File

@ -8,7 +8,7 @@ ENV DEBIAN_FRONTEND=noninteractive \
# For bitsandbytes
ENV BNB_GPU_TARGETS="gfx803;gfx900;gfx906;gfx908;gfx90a;gfx1010;gfx1030;gfx1100;gfx1101;gfx1102"
# For LLAMA
ENV GPU_TARGETS="gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101 gfx1102"
ENV GPU_TARGETS="gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101"
WORKDIR /tmp