diff --git a/README.md b/README.md index 8b6f53e..d21e7bc 100755 --- a/README.md +++ b/README.md @@ -1,6 +1,7 @@ -## tasknet : simple multi-task transformer fine-tuning with Trainer and HuggingFace datasets. -`tasknet` is an interface between Huggingface [datasets](https://huggingface.co/datasets) and Huggingface [Trainer](https://huggingface.co/docs/transformers/main_classes/trainer). +## tasknet : simple multi-task Trainer and HuggingFace datasets. +`tasknet` is an interface between Huggingface [datasets](https://huggingface.co/datasets) and Huggingface transformers [Trainer](https://huggingface.co/docs/transformers/main_classes/trainer). +Tasknet works with `transformers==4.34.1`. ## Task templates `tasknet` relies on task templates to avoid boilerplate codes. The task templates correspond to Transformers AutoClasses: