Skip to content

Commit

Permalink
fix: Update huggingface name for wrapped models
Browse files Browse the repository at this point in the history
  • Loading branch information
KennethEnevoldsen committed Jan 3, 2023
1 parent 9ac1320 commit cbb3f3b
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions src/dacy/hate_speech/wrapped_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
[hatespeech_detection.model]
@architectures = "spacy-wrap.ClassificationTransformerModel.v1"
name = "DaNLP/da-bert-hatespeech-detection"
name = "alexandrainst/da-bert-hatespeech-detection"
tokenizer_config = {"use_fast": true}
transformer_config = {}
mixed_precision = false
Expand All @@ -43,7 +43,7 @@
[hatespeech_classification.model]
@architectures = "spacy-wrap.ClassificationTransformerModel.v1"
name = "DaNLP/da-bert-hatespeech-classification"
name = "alexandrainst/da-bert-hatespeech-classification"
tokenizer_config = {"use_fast": true}
transformer_config = {}
mixed_precision = false
Expand Down
8 changes: 4 additions & 4 deletions src/dacy/sentiment/wrapped_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
[subjectivity.model]
@architectures = "spacy-wrap.ClassificationTransformerModel.v1"
name = "DaNLP/da-bert-tone-subjective-objective"
name = "alexandrainst/da-bert-tone-subjective-objective"
tokenizer_config = {"use_fast": true}
transformer_config = {}
mixed_precision = false
Expand All @@ -43,7 +43,7 @@
[polarity.model]
@architectures = "spacy-wrap.ClassificationTransformerModel.v1"
name = "DaNLP/da-bert-tone-sentiment-polarity"
name = "alexandrainst/da-bert-tone-sentiment-polarity"
tokenizer_config = {"use_fast": true}
transformer_config = {}
mixed_precision = false
Expand All @@ -66,7 +66,7 @@
[emotionally_laden.model]
@architectures = "spacy-wrap.ClassificationTransformerModel.v1"
name = "DaNLP/da-bert-emotion-binary"
name = "alexandrainst/da-bert-emotion-binary"
tokenizer_config = {"use_fast": true}
transformer_config = {}
mixed_precision = false
Expand All @@ -90,7 +90,7 @@
[emotion.model]
@architectures = "spacy-wrap.ClassificationTransformerModel.v1"
name = "DaNLP/da-bert-emotion-classification"
name = "alexandrainst/da-bert-emotion-classification"
tokenizer_config = {"use_fast": true}
transformer_config = {}
mixed_precision = false
Expand Down

0 comments on commit cbb3f3b

Please sign in to comment.