Upload requirements.txt
Browse files- 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).
|