Skip to content

GitLab

Unverified Commit c1d6fc84 authored by Muyang Li's avatar Muyang Li Committed by GitHub
Browse files

chore: use the official torch 2.7 with cuda 12.8 for the test (#332)

parent c374b2e3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
.github/workflows/pr_test_blackwell.yaml
View file @ c1d6fc84
......@@ -95,7 +95,7 @@ jobs:
which python
conda install -c conda-forge gxx=11 gcc=11
echo "Installing dependencies"
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
pip install ninja wheel diffusers transformers accelerate sentencepiece protobuf huggingface_hub
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment