Fetching metadata from the HF Docker repository...
fix: add flash attn
0e5ec8c
|
|
torch>=2.0.0 |
|
|
torchaudio>=2.0.0 |
|
|
transformers>=4.30.0 |
|
|
gradio>=4.0.0 |
|
|
numpy>=1.21.0 |
|
|
accelerate>=0.20.0 |
|
|
PyPDF2 |
|
|
beautifulsoup4 |
|
|
soundfile |
|
|
librosa |
|
|
tqdm |
|
|
requests |
|
|
openai |
|
|
PyYAML |
|
|
einops |
|
|
huggingface_hub |
|
|
flash-attn |
