Skip to content

Commit

Permalink
Add HF_TRANSFER to AMI (#729)
Browse files Browse the repository at this point in the history
Update install-huggingface-libraries.sh
  • Loading branch information
philschmid authored Nov 15, 2024
1 parent 459aa55 commit 78e5560
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion infrastructure/ami/scripts/install-huggingface-libraries.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,11 +14,14 @@ pip install --upgrade --no-cache-dir \
"notebook==7.0.6" \
"markupsafe==2.1.1" \
"jinja2==3.1.2" \
"attrs==23.1.0"
"attrs==23.1.0" \
"hf_transfer>=0.1.4"

# Temporary fix for the issue: https://github.com/huggingface/optimum-neuron/issues/142
pip install -U optimum
echo 'export PATH="${HOME}/.local/bin:$PATH"' >> "${HOME}/.bashrc"
# Add HF_TRANSFER by default
echo 'export HF_HUB_ENABLE_HF_TRANSFER=1' >> "${HOME}/.bashrc"

echo "Step: install-and-copy-optimum-neuron-examples"
git clone -b $OPTIMUM_VERSION https://github.com/huggingface/optimum-neuron.git
Expand Down

0 comments on commit 78e5560

Please sign in to comment.