satkinson commited on
Commit
8eb5821
1 Parent(s): 8864763

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -10,7 +10,7 @@ metrics:
10
 
11
  ## DialoGPT model fine-tuned using Amazon's Topical Chat Dataset
12
 
13
- This model is fine-tuned from the original [DialoGPT-medium](https://huggingface.co/microsoft/DialoGPT-medium).
14
 
15
  This model was fine-tuned on a subset of messages from [Amazon's Topical Chat dataset](https://www.kaggle.com/arnavsharmaas/chatbot-dataset-topical-chat) (due to processing limitations, I restricted my training data to 50,000 observations.
16
 
 
10
 
11
  ## DialoGPT model fine-tuned using Amazon's Topical Chat Dataset
12
 
13
+ This model is fine-tuned from the original [DialoGPT-small](https://huggingface.co/microsoft/DialoGPT-small).
14
 
15
  This model was fine-tuned on a subset of messages from [Amazon's Topical Chat dataset](https://www.kaggle.com/arnavsharmaas/chatbot-dataset-topical-chat) (due to processing limitations, I restricted my training data to 50,000 observations.
16