adidsh commited on
Commit
83198dc
·
verified ·
1 Parent(s): 7667a87

Upload requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +19 -0
requirements.txt ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # The model code ships in this repo; these are its only third-party dependencies.
2
+ #
3
+ # Install torch and torchvision TOGETHER, from the index for your CUDA version -- torchvision pins
4
+ # an exact torch, so resolving them separately is how you end up with a mismatched pair:
5
+ #
6
+ # pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
7
+ # pip install -r requirements.txt
8
+ #
9
+ # ./install.sh does this for you.
10
+
11
+ torch>=2.4 # transformers 5.7 floor
12
+ torchvision # stage 2 only; MUST match torch (see above)
13
+ transformers>=5.7 # PPDocLayoutV3
14
+ accelerate>=1.1
15
+ numpy>=1.17
16
+ pillow>=10.0.1
17
+ huggingface_hub>=1.0
18
+
19
+ # scipy is NOT needed for inference. Finetuning the detector needs it (the Hungarian matcher).