[CI] Fix huggingface access for test_flash_attention_4.py (#12846)
This commit is contained in:
@@ -22,8 +22,6 @@ echo "CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES:-}"
|
||||
|
||||
# Clear torch compilation cache
|
||||
python3 -c 'import os, shutil, tempfile, getpass; cache_dir = os.environ.get("TORCHINDUCTOR_CACHE_DIR") or os.path.join(tempfile.gettempdir(), "torchinductor_" + getpass.getuser()); shutil.rmtree(cache_dir, ignore_errors=True)'
|
||||
rm -rf /root/.cache/flashinfer
|
||||
pip3 uninstall flashinfer-python flashinfer-cubin flashinfer-jit-cache || true
|
||||
|
||||
# Install apt packages
|
||||
apt install -y git libnuma-dev libssl-dev pkg-config
|
||||
@@ -80,6 +78,7 @@ if [ "$IS_BLACKWELL" = "1" ]; then
|
||||
|
||||
# Clean up existing installations
|
||||
$PIP_CMD uninstall -y sgl-kernel sglang $PIP_INSTALL_SUFFIX || true
|
||||
$PIP_CMD uninstall -y flashinfer-python flashinfer-cubin flashinfer-jit-cache $PIP_INSTALL_SUFFIX || true
|
||||
else
|
||||
# In normal cases, we use uv, which is much faster than pip.
|
||||
pip install --upgrade pip
|
||||
@@ -91,6 +90,7 @@ else
|
||||
|
||||
# Clean up existing installations
|
||||
$PIP_CMD uninstall sgl-kernel sglang || true
|
||||
$PIP_CMD uninstall flashinfer-python flashinfer-cubin flashinfer-jit-cache || true
|
||||
fi
|
||||
|
||||
# Install the main package
|
||||
|
||||
Reference in New Issue
Block a user