Minor fixes

README.md

@@ -46,7 +46,6 @@ Inputs require SELFIES tokenization **and** a precomputed distance matrix
 (`relative_position`). Use the helper bundled in the repo:
 
 ```python
-
 tokenizer = AutoTokenizer.from_pretrained("IlPakoZ/m5-encoder", trust_remote_code=True)
 
 smiles = "CCO"

modeling_m5_encoder.py

@@ -36,6 +36,7 @@ class M5EncoderConfig(T5Config):
         relative_attention_max_distance=128,
         relative_attention_num_buckets=48,
         vocab_size=1032,
+        num_decoder_layers=0,
         **kwargs,
     ):
         super().__init__(d_ff=d_ff, 
@@ -50,7 +51,7 @@ class M5EncoderConfig(T5Config):
                          relative_attention_max_distance=relative_attention_max_distance, 
                          relative_attention_num_buckets=relative_attention_num_buckets, 
                          vocab_size=vocab_size,
-                         num_decoder_layers=0,
+                         num_decoder_layers=num_decoder_layers,
                          **kwargs)
 
 class M5Encoder(PreTrainedModel):