fix docker flashattn
This commit is contained in:
1 parent
6f63050e1b
commit
e3141f5f1b
1 file changed
+1
@@ -4,6 +4,7 @@ FROM nvcr.io/nvidia/pytorch:24.02-py3
|
||||
|
||||
# Define environments
|
||||
ENV MAX_JOBS=4
|
||||
ENV FLASH_ATTENTION_FORCE_BUILD=TRUE
|
||||
|
||||
# Define installation arguments
|
||||
ARG INSTALL_BNB=false
|
||||
|
||||
Reference in new issue
Block a user