diff --git a/.github/workflows/benchmark_torch_compile_nightly.yml b/.github/workflows/benchmark_torch_compile_nightly.yml index 6857b79f68..1274fb929a 100644 --- a/.github/workflows/benchmark_torch_compile_nightly.yml +++ b/.github/workflows/benchmark_torch_compile_nightly.yml @@ -27,10 +27,10 @@ jobs: cd .. pwd rm -rf _tool - - name: Setup Python 3.8 + - name: Setup Python 3.10 uses: actions/setup-python@v4 with: - python-version: 3.8 + python-version: 3.10 architecture: x64 - name: Setup Java 17 uses: actions/setup-java@v3 @@ -46,8 +46,8 @@ jobs: sudo apt-get update -y sudo apt-get install -y apache2-utils pip install -r benchmarks/requirements-ab.txt - chmod +x install_segment_anything_fast.sh - source install_segment_anything_fast.sh + chmod +x examples/large_models/segment_anything_fast/install_segment_anything_fast.sh + source examples/large_models/segment_anything_fast/install_segment_anything_fast.sh export SEGMENT_ANYTHING_FAST_USE_FLASH_4=0 - name: Benchmark gpu nightly run: python benchmarks/auto_benchmark.py --input benchmarks/benchmark_config_torch_compile_gpu.yaml --skip false --nightly True