diff --git a/airootfs/root/customize_airootfs/scripts/1000-axolotl-dependencies.sh b/airootfs/root/customize_airootfs/scripts/1000-axolotl-dependencies.sh index a15c1cd..beb6d2f 100644 --- a/airootfs/root/customize_airootfs/scripts/1000-axolotl-dependencies.sh +++ b/airootfs/root/customize_airootfs/scripts/1000-axolotl-dependencies.sh @@ -15,7 +15,7 @@ pushd "axolotl" pip3 install packaging wheel # install flash-attn without build isolation - cat requirements.txt | grep flash-attn | xargs pip install --no-build-isolation + cat requirements.txt | grep flash-attn | xargs pip3 install --no-build-isolation # install axolotl pip3 install -e ".[flash-attn,deepspeed]"