From f6e0a11b18cb70c4b1e102334d8dabccbafc8983 Mon Sep 17 00:00:00 2001 From: MekkCyber Date: Wed, 25 Dec 2024 11:31:32 +0000 Subject: [PATCH] fix --- docker/transformers-quantization-latest-gpu/Dockerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/transformers-quantization-latest-gpu/Dockerfile b/docker/transformers-quantization-latest-gpu/Dockerfile index bde0abc8b5a970..7d129405792d51 100755 --- a/docker/transformers-quantization-latest-gpu/Dockerfile +++ b/docker/transformers-quantization-latest-gpu/Dockerfile @@ -68,11 +68,11 @@ RUN git clone https://github.com/casper-hansen/AutoAWQ.git && cd AutoAWQ && pip RUN python3 -m pip install --no-cache-dir optimum-quanto # Add eetq for quantization testing -RUN python3 -m pip install git+https://github.com/NetEase-FuXi/EETQ.git +# RUN python3 -m pip install git+https://github.com/NetEase-FuXi/EETQ.git # Add flute-kernel and fast_hadamard_transform for quantization testing -RUN python3 -m pip install --no-cache-dir flute-kernel==0.3.0 -i https://flute-ai.github.io/whl/cu118 -RUN python3 -m pip install --no-cache-dir fast_hadamard_transform==1.0.4.post1 +# RUN python3 -m pip install --no-cache-dir flute-kernel==0.3.0 -i https://flute-ai.github.io/whl/cu118 +# RUN python3 -m pip install --no-cache-dir fast_hadamard_transform==1.0.4.post1 # When installing in editable mode, `transformers` is not recognized as a package. # this line must be added in order for python to be aware of transformers.