wjs0725 commited on
Commit
ba5c0fa
1 Parent(s): dab49c5

Upload requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -0
requirements.txt ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ torch
2
+ einops
3
+ accelerate==0.34.2
4
+ deepspeed==0.14.4
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