daoyuan98 commited on
Commit
17ceb32
·
verified ·
1 Parent(s): 262bee5

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +17 -1
requirements.txt CHANGED
@@ -1 +1,17 @@
1
- einops
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ torch==2.3.1
2
+ accelerate==0.34.2
3
+ deepspeed==0.14.4
4
+ einops==0.8.0
5
+ transformers==4.41.2
6
+ huggingface-hub==0.24.6
7
+ datasets
8
+ omegaconf
9
+ diffusers
10
+ sentencepiece
11
+ opencv-python
12
+ matplotlib
13
+ onnxruntime
14
+ torchvision
15
+ timm
16
+ invisible-watermark
17
+ fire