balhafni commited on
Commit
2750503
1 Parent(s): f049c18

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -6,7 +6,8 @@ language:
6
 
7
  # Personalized Text Generation with Fine-Grained Linguistic Control
8
 
9
- ## Model Description
 
10
 
11
 
12
  ## Usage
 
6
 
7
  # Personalized Text Generation with Fine-Grained Linguistic Control
8
 
9
+ This model was obtained by fine-tuning the [Pythia-1b](https://huggingface.co/EleutherAI/) model on three datasets: [Blogs Authorship Corpus](https://u.cs.biu.ac.il/~koppel/BlogCorpus.htm), [IMDb62](https://umlt.infotech.monash.edu/?page_id=266), and [Amazon 5-core Reviews](https://cseweb.ucsd.edu/~jmcauley/datasets/amazon_v2/).
10
+ The model was built so that it can generate text based on multiple fine-grained linguistic attributes that reflect authors’ writing style. Our fine-tuning procedure and the hyperparameters we used can be found in our paper "*[Personalized Text Generation with Fine-Grained Linguistic Control](https://arxiv.org/abs/2402.04914)*" Our fine-tuning code and data can be found [here](https://github.com/balhafni/personalized-gen).
11
 
12
 
13
  ## Usage