Skip to content

GitLab

Commit f434bfc6 authored by Julien Chaumond's avatar Julien Chaumond
Browse files

[pplm] Update S3 links 


Co-Authored-By: default avatarPiero Molino <w4nderlust@gmail.com>
parent 96e83506
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
examples/pplm/run_pplm.py
View file @ f434bfc6
......@@ -59,7 +59,7 @@ BAG_OF_WORDS_ARCHIVE_MAP = {
DISCRIMINATOR_MODELS_PARAMS = {
"clickbait": {
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/clickbait_classifierhead.pt",
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/clickbait_classifier_head.pt",
"class_size": 2,
"embed_size": 1024,
"class_vocab": {"non_clickbait": 0, "clickbait": 1},
......@@ -67,7 +67,7 @@ DISCRIMINATOR_MODELS_PARAMS = {
"pretrained_model": "gpt2-medium",
},
"sentiment": {
"url": "http://s.yosinski.com/SST_classifier_head.pt",
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/SST_classifier_head.pt",
"class_size": 5,
"embed_size": 1024,
"class_vocab": {"very_positive": 2, "very_negative": 3},
......@@ -75,7 +75,7 @@ DISCRIMINATOR_MODELS_PARAMS = {
"pretrained_model": "gpt2-medium",
},
"toxicity": {
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/toxicity_classifierhead.pt",
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/toxic_classifier_head.pt",
"class_size": 2,
"embed_size": 1024,
"class_vocab": {"non_toxic": 0, "toxic": 1},
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment