Skip to content

redpin-pankaj/Emotion-Emphasis

Repository files navigation

Setup Guide

Step 1: Download DialoGPT fine-tuned model weights

Download the fine-tuned model weights from the URL https://convaisharables.blob.core.windows.net/lsp/multiref/medium_ft.pkl and save it in current(Emotion-Emphasis) directory. The filename of the downloaded file should be medium_ft.pkl

Step 2: Save the model as PyTorch checkpoint

Run the create_checkpoint_dialogpt_ft_model.py file to save the model as PyTorch checkpoint. Please refer to https://huggingface.co/transformers/serialization.html#serialization-best-practices for more details.

Step 3: Download pre-trained TorchMoji mode

Run the script scripts/download_weights.py which will download the pre-trained TorchMoji weights into model folder. Type y (yes) when prompted to confirm download.

Step 4: Run DialogueAgent.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages