Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.
12 changes: 5 additions & 7 deletions .github/workflows/pull.yml
Original file line number Diff line number Diff line change
Expand Up @@ -904,14 +904,12 @@ jobs:
echo "et-git-hash=$(cat ${TORCHCHAT_ROOT}/install/.pins/et-pin.txt)" >> "$GITHUB_ENV"
- name: Load or install ET
id: install-et
uses: actions/cache@v3
env:
cache-key: et-build-${{runner.os}}-${{runner.arch}}-${{env.et-git-hash}}
uses: actions/cache@v4
with:
path: ./et-build
key: ${{env.cache-key}}
restore-keys: |
${{env.cache-key}}
path: |
./et-build
./torchchat/utils/scripts
key: et-build-${{runner.os}}-${{runner.arch}}-${{env.et-git-hash}}-${{ hashFiles('**/install_et.sh') }}
- if: ${{ steps.install-et.outputs.cache-hit != 'true' }}
continue-on-error: true
run: |
Expand Down
2 changes: 1 addition & 1 deletion install/install_requirements.sh
Original file line number Diff line number Diff line change
Expand Up @@ -103,5 +103,5 @@ fi

(
set -x
$PIP_EXECUTABLE install lm-eval=="0.4.2"
$PIP_EXECUTABLE install evaluate=="0.4.3" lm-eval=="0.4.2" psutil=="6.0.0"
)
4 changes: 3 additions & 1 deletion torchchat/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,9 @@
SequenceParallel,
)
from torch.nn import functional as F

# TODO: remove this after we figure out where in torchtune an `evaluate` module
# is being imported, which is being confused with huggingface's `evaluate``.
import lm_eval # noqa
from torchtune.models.clip import clip_vision_encoder
from torchtune.models.llama3_1._component_builders import llama3_1 as llama3_1_builder
from torchtune.models.llama3_2_vision._component_builders import (
Expand Down
5 changes: 5 additions & 0 deletions torchchat/utils/scripts/install_et.sh
Original file line number Diff line number Diff line change
Expand Up @@ -20,4 +20,9 @@ find_cmake_prefix_path
clone_executorch
install_executorch_libs $ENABLE_ET_PYBIND
install_executorch_python_libs $ENABLE_ET_PYBIND
# TODO: figure out the root cause of 'AttributeError: module 'evaluate'
# has no attribute 'utils'' error from evaluate CI jobs and remove
# `import lm_eval` from torchchat.py since it requires a specific version
# of numpy.
pip install numpy=='1.26.4'
popd
Loading