1inkusFace commited on
Commit
52b4a6e
·
verified ·
1 Parent(s): c8ff766

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -5,7 +5,7 @@ transformers
5
  accelerate
6
  numpy
7
  --extra-index-url https://download.pytorch.org/whl/cu124
8
- torch
9
  huggingface_hub[hf_transfer]
10
  sentencepiece
11
  torchvision #==0.20.1
@@ -20,7 +20,7 @@ git+https://github.com/huggingface/diffusers.git@464374fb87610c53b2cf81e08d3df62
20
  #git+https://github.com/huggingface/diffusers.git
21
  git+https://github.com/Howe2018/ParaAttention.git
22
  torchao #==0.7.0
23
- imageio-ffmpeg #==0.5.1
24
  nvtx #==0.2.10
25
- opencv-python #==4.10.0.84
26
- imageio #==2.36.1
 
5
  accelerate
6
  numpy
7
  --extra-index-url https://download.pytorch.org/whl/cu124
8
+ torch==2.5.1
9
  huggingface_hub[hf_transfer]
10
  sentencepiece
11
  torchvision #==0.20.1
 
20
  #git+https://github.com/huggingface/diffusers.git
21
  git+https://github.com/Howe2018/ParaAttention.git
22
  torchao #==0.7.0
23
+ imageio-ffmpeg==0.5.1
24
  nvtx #==0.2.10
25
+ opencv-python==4.10.0.84
26
+ imageio==2.36.1