| # requirements.txt | |
| torch==2.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html | |
| transformers==4.46.3 | |
| einops | |
| addict | |
| easydict | |
| Pillow | |
| gradio | |
| # skip vllm and flash-attn if GPU is required | |
| # requirements.txt | |
| torch==2.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html | |
| transformers==4.46.3 | |
| einops | |
| addict | |
| easydict | |
| Pillow | |
| gradio | |
| # skip vllm and flash-attn if GPU is required | |