| # UI and PDF tools | |
| gradio | |
| PyMuPDF | |
| # --- Core Model Dependencies with Forced Versions --- | |
| # Forcing newer versions to fix the 'LlamaFlashAttention2' ImportError | |
| transformers>=4.40.0 | |
| accelerate>=0.28.0 | |
| torch>=2.1.0 | |
| torchvision>=0.16.0 | |
| # Other model dependencies | |
| sentencepiece | |
| Pillow | |
| einops | |
| easydict | |
| addict | |
| matplotlib |