Task
Other
J
jinaai/jina-bert-flash-implementation
Other·jinaai· 607.1K· 5
transformers
region:eu
BERT with Flash-Attention Installing dependencies To run the model on GPU, you need to install Flash Attention. You may either install from pypi (which may not work with fused-dense), or from source. To install from source, clone the GitHub repository: console git clone git@github.com:Dao-AILab/flash-attention.git
# pull & run locally
pip install mlforge-sdk && mlforge pull jinaai/jina-bert-flash-implementation
pip install mlforge-sdk && mlforge pull jinaai/jina-bert-flash-implementation
Model details
Provider
jinaai
Framework
transformers
Downloads
607.1K
Likes
5
Updated
2024-05-31
About jinaai/jina-bert-flash-implementation
BERT with Flash-Attention Installing dependencies To run the model on GPU, you need to install Flash Attention. You may either install from pypi (which may not work with fused-dense), or from source. To install from source, clone the GitHub repository: