Update README.md
Browse files
README.md
CHANGED
|
@@ -10,7 +10,7 @@ tags:
|
|
| 10 |
---
|
| 11 |
# Note:
|
| 12 |
This model is copied version of mRNABERT which removes the FlashAttention integration with Trition. This allows the model to be installed off HuggingFace without having to uninstall Triton. Running the below example code yields identical output compared to the original verison.
|
| 13 |
-
```
|
| 14 |
import torch
|
| 15 |
from transformers import AutoTokenizer, AutoModel
|
| 16 |
from transformers.models.bert.configuration_bert import BertConfig
|
|
|
|
| 10 |
---
|
| 11 |
# Note:
|
| 12 |
This model is copied version of mRNABERT which removes the FlashAttention integration with Trition. This allows the model to be installed off HuggingFace without having to uninstall Triton. Running the below example code yields identical output compared to the original verison.
|
| 13 |
+
```python
|
| 14 |
import torch
|
| 15 |
from transformers import AutoTokenizer, AutoModel
|
| 16 |
from transformers.models.bert.configuration_bert import BertConfig
|