IlPakoZ commited on
Commit
9db47da
·
1 Parent(s): 3e2a3e1

Edited README.md

Browse files
Files changed (2) hide show
  1. README.md +4 -0
  2. 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