Prompt_tuning_T5_mT5

Instructions to run code

-> Run the python notebook PrefixTuningUsingT5andMT5.ipynb by uploading it to google colab or jupyter notebook

-> Follow the steps given in the notebook

-> All packages that needs to be installed are mentioned in the PrefixTuningUsingT5andMT5.ipynb notebook

To run T5

-> set model_name_or_path variable to 't5-small'

To run mT5

-> set model_name_or_path variable to 'google/mt5-base'

Link to Trained Model pickle file

anandanne / Prompt_tuning_T5_mT5