Commit e6ca2021 authored by ingyuseong's avatar ingyuseong
Browse files

Resolve confilict on init file

parent dae050b4
......@@ -57,6 +57,7 @@ from . import klue
from . import ko_translation
from . import korquad
from . import korunsmile
from . import kohatespeech
########################################
# Translation tasks
......@@ -69,7 +70,6 @@ gpt3_translation_benchmarks = {
"wmt16": ['en-ro', 'ro-en', 'de-en', 'en-de'], # German, Romanian
}
# 28 total
selected_translation_benchmarks = {
**gpt3_translation_benchmarks,
......@@ -88,7 +88,7 @@ all_translation_benchmarks = {
# All tasks
########################################
# task 레지스트리에 클래스 추가
TASK_REGISTRY = {
# GLUE
"cola": glue.CoLA,
......@@ -322,9 +322,11 @@ TASK_REGISTRY = {
"ko_en_translation": ko_translation.KoEnTranslation,
"en_ko_translation": ko_translation.EnKoTranslation,
"korunsmile": korunsmile.KorUnSmile
"kohatespeech":kohatespeech.HateSpeech,
"kohatespeech_gen_bias":kohatespeech.GenderBias,
"kohatespeech_apeach":kohatespeech.Apeach
}
ALL_TASKS = sorted(list(TASK_REGISTRY))
......
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