ahczhg commited on
Commit
09393ef
·
verified ·
1 Parent(s): 181a8ac

Upload requirements.txt with huggingface_hub

Browse files
Files changed (1) hide show
  1. requirements.txt +39 -0
requirements.txt ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # DeepSeek-OCR Colab Requirements
2
+ # For local setup only - Google Colab already includes most of these
3
+
4
+ # Core dependencies
5
+ torch>=2.0.0
6
+ torchvision>=0.15.0
7
+ torchaudio>=2.0.0
8
+
9
+ # Transformers and tokenizers
10
+ transformers==4.46.3
11
+ tokenizers==0.20.3
12
+
13
+ # PDF and image processing
14
+ PyMuPDF>=1.23.0
15
+ img2pdf>=0.5.0
16
+ Pillow>=10.0.0
17
+
18
+ # Utilities
19
+ einops>=0.7.0
20
+ easydict>=1.10
21
+ addict>=2.4.0
22
+ numpy>=1.24.0
23
+
24
+ # Flash Attention (optional, may require compilation)
25
+ # flash-attn==2.7.3
26
+
27
+ # Additional dependencies
28
+ safetensors>=0.4.1
29
+ huggingface-hub>=0.23.2
30
+
31
+ # For Jupyter notebooks (local)
32
+ jupyter>=1.0.0
33
+ ipywidgets>=8.0.0
34
+
35
+ # Note: Flash Attention installation can be challenging and may require:
36
+ # - CUDA Toolkit 11.8 or 12.1+
37
+ # - Matching PyTorch CUDA version
38
+ # - C++ compiler
39
+ # If flash-attn fails to install, the notebook will fall back to standard attention