phamhieu2001 commited on
Commit
7b4363f
·
verified ·
1 Parent(s): f429425

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +10 -5
requirements.txt CHANGED
@@ -1,6 +1,11 @@
 
 
 
1
  accelerate
2
- diffusers
3
- invisible_watermark
4
- torch
5
- transformers
6
- xformers
 
 
 
1
+ torch==2.6.0
2
+ transformers==4.46.3
3
+ tokenizers==0.20.3
4
  accelerate
5
+ einops
6
+ addict
7
+ easydict
8
+ torchvision
9
+ flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
10
+ PyMuPDF
11
+ hf_transfer