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 = { ...@@ -59,7 +59,7 @@ BAG_OF_WORDS_ARCHIVE_MAP = {
DISCRIMINATOR_MODELS_PARAMS = { DISCRIMINATOR_MODELS_PARAMS = {
"clickbait": { "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, "class_size": 2,
"embed_size": 1024, "embed_size": 1024,
"class_vocab": {"non_clickbait": 0, "clickbait": 1}, "class_vocab": {"non_clickbait": 0, "clickbait": 1},
...@@ -67,7 +67,7 @@ DISCRIMINATOR_MODELS_PARAMS = { ...@@ -67,7 +67,7 @@ DISCRIMINATOR_MODELS_PARAMS = {
"pretrained_model": "gpt2-medium", "pretrained_model": "gpt2-medium",
}, },
"sentiment": { "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, "class_size": 5,
"embed_size": 1024, "embed_size": 1024,
"class_vocab": {"very_positive": 2, "very_negative": 3}, "class_vocab": {"very_positive": 2, "very_negative": 3},
...@@ -75,7 +75,7 @@ DISCRIMINATOR_MODELS_PARAMS = { ...@@ -75,7 +75,7 @@ DISCRIMINATOR_MODELS_PARAMS = {
"pretrained_model": "gpt2-medium", "pretrained_model": "gpt2-medium",
}, },
"toxicity": { "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, "class_size": 2,
"embed_size": 1024, "embed_size": 1024,
"class_vocab": {"non_toxic": 0, "toxic": 1}, "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