oyly commited on
Commit
154e0f2
·
1 Parent(s): 091540f

align requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +10 -14
requirements.txt CHANGED
@@ -1,19 +1,15 @@
1
  pydantic==2.10.6
2
- torch
3
- einops
4
- accelerate==0.34.2
5
- einops==0.8.0
6
- transformers==4.41.2
7
- huggingface-hub==0.24.6
8
- datasets
9
- omegaconf
10
- diffusers
11
- sentencepiece
12
  opencv-python
13
  matplotlib
14
  onnxruntime
15
- torchvision
16
- timm
17
- invisible-watermark
18
  fire
19
- tqdm
 
 
1
  pydantic==2.10.6
2
+ torch==2.6.0
3
+ einops==0.8.1
4
+ transformers==4.49.0
5
+ huggingface-hub==0.34.4
6
+ omegaconf==2.3.0
7
+ sentencepiece==0.2.0
 
 
 
 
8
  opencv-python
9
  matplotlib
10
  onnxruntime
11
+ torchvision==0.21.0
12
+ invisible-watermark==0.2.0
 
13
  fire
14
+ tqdm
15
+ numpy==2.2.4