Skip to content

Commit ba5c1c2

Browse files
committedFeb 10, 2024
fixed AWQ to remain on older lm-eval
1 parent 99a3a1f commit ba5c1c2

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed
 

‎packages/llm/awq/Dockerfile

+3
Original file line numberDiff line numberDiff line change
@@ -37,5 +37,8 @@ COPY quantize.py awq/
3737

3838
WORKDIR /
3939

40+
# No module named 'lm_eval.base'
41+
RUN pip3 install --no-cache-dir --verbose 'lm-eval<=0.3.0'
42+
4043
# make sure it loads
4144
RUN pip3 show awq && python3 -c 'import awq' && python3 -m awq.entry --help

0 commit comments

Comments
 (0)
Please sign in to comment.