PahaII commited on
Commit
42cf800
1 Parent(s): 777c9ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -48,7 +48,7 @@ We use the prompt template `Publisher: {vox} article: ` for training. We trained
48
 
49
  >>> publisher = "Reuters"
50
  >>> assert publisher in ["Reuters", "NYT", "CNBC", "Hill", "People", "CNN", "Vice", "Mashable", "Refinery", "BI", "TechCrunch", "Verge", "TMZ", "Axios", "Vox", "Guardian", "BBCNews", "WashingtonPost", "USAToday"]
51
- >>> prompt = f"Publisher: {publisher.lower()} article: Local police is dealing with a car accident"
52
 
53
  >>> inputs = tokenizer(prompt, return_tensors="pt")
54
  >>> out = model.generate(**inputs, penalty_alpha=0.6)
 
48
 
49
  >>> publisher = "Reuters"
50
  >>> assert publisher in ["Reuters", "NYT", "CNBC", "Hill", "People", "CNN", "Vice", "Mashable", "Refinery", "BI", "TechCrunch", "Verge", "TMZ", "Axios", "Vox", "Guardian", "BBCNews", "WashingtonPost", "USAToday"]
51
+ >>> prompt = f"{tokenizer.bos_token}Publisher: {publisher.lower()} article: Local police is dealing with a car accident"
52
 
53
  >>> inputs = tokenizer(prompt, return_tensors="pt")
54
  >>> out = model.generate(**inputs, penalty_alpha=0.6)