add peft dependencies in dockerfile

flexflow · Oct 31, 2023 · eb9e2b8 · eb9e2b8
1 parent cdc12e6
commit eb9e2b8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docker/flexflow-environment/Dockerfile b/docker/flexflow-environment/Dockerfile
@@ -95,6 +95,8 @@ RUN conda install -c conda-forge cmake make pillow cmake-build-extension pybind1
 RUN conda install pytorch torchvision torchaudio -c pytorch
 RUN conda install -c conda-forge onnx transformers>=4.31.0 sentencepiece einops
 RUN pip3 install tensorflow notebook
+# PEFT-related
+RUN pip3 install scipy bitsandbytes datasets accelerate loralib peft
 
 # Install Rust
 RUN curl https://sh.rustup.rs -sSf | sh -s -- -y