Update tasks comment

config.json

@@ -23,7 +23,7 @@
   "relative_attention_max_distance": 128,
   "relative_attention_num_buckets": 32,
   "tie_word_embeddings": false,
-  "transformers_version": "4.17.0",
+  "transformers_version": "4.18.0",
   "use_cache": true,
   "vocab_size": 32128
 }

tasks.py

@@ -48,7 +48,7 @@ def target_to_key(x, key_map, target_key):
     return {**key_map, target_key: x}
 
 
-# Final pretraining task used in Raffel et al., 2019 adaptated to NCC
+# Final pretraining task used in Raffel et al., 2019 adaptated to our use
 dataset_name = "/researchdisk/lm_training_dataset_full"
 dataset_params = {"from_disk_path": dataset_name}