Edited README.md
Browse files- README.md +4 -0
- modeling_m5_encoder.py +0 -1
README.md
CHANGED
|
@@ -21,6 +21,10 @@ distance-aware relative position encodings. Two classes are available:
|
|
| 21 |
The model is pretrained on multi-task regression tasks, including quantum chemistry (QC) tasks
|
| 22 |
from the [PubChemQC B3LYP/PM6 dataset](https://nakatamaho.riken.jp/pubchemqc.riken.jp/b3lyp_pm6_datasets.html).
|
| 23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
## Usage
|
| 25 |
|
| 26 |
```python
|
|
|
|
| 21 |
The model is pretrained on multi-task regression tasks, including quantum chemistry (QC) tasks
|
| 22 |
from the [PubChemQC B3LYP/PM6 dataset](https://nakatamaho.riken.jp/pubchemqc.riken.jp/b3lyp_pm6_datasets.html).
|
| 23 |
|
| 24 |
+
## Requirements
|
| 25 |
+
|
| 26 |
+
**This model was tested and implemented with Transformers version 4.51.3, so issues might appear in other versions.**
|
| 27 |
+
|
| 28 |
## Usage
|
| 29 |
|
| 30 |
```python
|
modeling_m5_encoder.py
CHANGED
|
@@ -12,7 +12,6 @@ from transformers.cache_utils import DynamicCache, EncoderDecoderCache
|
|
| 12 |
from transformers.models.t5.configuration_t5 import T5Config
|
| 13 |
from transformers.modeling_outputs import BaseModelOutputWithPastAndCrossAttentions, BaseModelOutput
|
| 14 |
from transformers.utils import DUMMY_INPUTS, DUMMY_MASK, is_torch_fx_proxy, is_torchdynamo_compiling
|
| 15 |
-
from transformers.modeling_outputs import BaseModelOutputWithPastAndCrossAttentions, BaseModelOutput
|
| 16 |
from transformers.utils.deprecation import deprecate_kwarg
|
| 17 |
from .common import M5Pooler
|
| 18 |
from .prepare_data import get_positional_encodings_and_align
|
|
|
|
| 12 |
from transformers.models.t5.configuration_t5 import T5Config
|
| 13 |
from transformers.modeling_outputs import BaseModelOutputWithPastAndCrossAttentions, BaseModelOutput
|
| 14 |
from transformers.utils import DUMMY_INPUTS, DUMMY_MASK, is_torch_fx_proxy, is_torchdynamo_compiling
|
|
|
|
| 15 |
from transformers.utils.deprecation import deprecate_kwarg
|
| 16 |
from .common import M5Pooler
|
| 17 |
from .prepare_data import get_positional_encodings_and_align
|