Ketengan-Diffusion-Lab commited on
Commit
82daaef
1 Parent(s): 2a63ac0

Update pre-requirements.txt

Browse files
Files changed (1) hide show
  1. pre-requirements.txt +3 -0
pre-requirements.txt CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  pip==24.2
2
  ninja
3
  setuptools
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.3.1+cu118
3
+ torchvision==0.18.1+cu118
4
  pip==24.2
5
  ninja
6
  setuptools