mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-01 11:12:50 +08:00
parent
b3e4793ded
commit
93ba3bd5b0
@ -5,6 +5,7 @@ FROM nvcr.io/nvidia/pytorch:24.02-py3
|
||||
# Define environments
|
||||
ENV MAX_JOBS=4
|
||||
ENV FLASH_ATTENTION_FORCE_BUILD=TRUE
|
||||
ENV VLLM_WORKER_MULTIPROC_METHOD=spawn
|
||||
|
||||
# Define installation arguments
|
||||
ARG INSTALL_BNB=false
|
||||
|
Loading…
x
Reference in New Issue
Block a user