Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
camenduru authored Dec 6, 2024
1 parent 5da8354 commit 8ff9f20
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions TRELLIS_jupyter.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@
},
"outputs": [],
"source": [
"!apt install aria2 -qqy \n",
"\n",
"!pip install torch==2.5.0+cu124 torchvision==0.20.0+cu124 torchaudio==2.5.0+cu124 torchtext==0.18.0 torchdata==0.8.0 --extra-index-url https://download.pytorch.org/whl/cu124\n",
"!pip install xformers==0.0.28.post2\n",
"!pip install https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl\n",
Expand Down

0 comments on commit 8ff9f20

Please sign in to comment.