wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-08-07 11:17:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Store Message Toxicity in database (#553)

Browse Source 
* [NEW] MessageToxicity table

* [NEW] Alembic message Toxicity

* [NEW] Model name enum

* [NEW] Refactor Enum HF

* [NEW] Settings: DEBUT_SKIP_TOXICITY_CALCULATION

* [NEW] Store toxicity values

* [FIX] Merge conflict

* [FIX] Documentation

* [NEW] save_toxicity: function

* [FIX] Formatted string

* [NEW] DEBUG_SKIP_TOXICITY_CALCULATION=True

* [FIX] HfClassificationModel

* [FIX] Alembic merge heads

* [NEW] Refactor save_toxicity

* [NEW] Separating score/label

* [NEW] Store score and label

* [FIX] Cleaning Alembic

* [NEW] Clean HF names

* [NEW] Not type hinting

* [NEW] Update alembic versions

* [NEW] Revert the changes

* [NEW] Type hinting label & score

* Updated down_revision in migration script

Co-authored-by: Andreas Köpf <andreas.koepf@xamla.com>
This commit is contained in:
Nil Andreu
2023-01-14 12:22:55 +00:00
committed by GitHub
co-authored by Andreas Köpf
parent dbf8f77072
commit a902c600fa
11 changed files with 119 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker-compose.yaml
+1
View File
@@ -101,6 +101,7 @@ services:
- DEBUG_USE_SEED_DATA=True
- DEBUG_ALLOW_SELF_LABELING=True
- MAX_WORKERS=1
- DEBUG_SKIP_TOXICITY_CALCULATION=True
- DEBUG_SKIP_EMBEDDING_COMPUTATION=True
depends_on:
db: