Skip to content

Commit 408ad40

Browse files
MekkCyberzucchini-nlp
authored andcommitted
Disable kernels for quantization (huggingface#37446)
fix
1 parent 803f7fd commit 408ad40

File tree

1 file changed

+2
-0
lines changed
  • docker/transformers-quantization-latest-gpu

1 file changed

+2
-0
lines changed

docker/transformers-quantization-latest-gpu/Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@ SHELL ["sh", "-lc"]
1212
ARG PYTORCH='2.6.0'
1313
# Example: `cu102`, `cu113`, etc.
1414
ARG CUDA='cu121'
15+
# Disable kernel mapping for quantization tests
16+
ENV DISABLE_KERNEL_MAPPING=1
1517

1618
RUN apt update
1719
RUN apt install -y git libsndfile1-dev tesseract-ocr espeak-ng python3 python3-pip ffmpeg

0 commit comments

Comments
 (0)