from local

This commit is contained in:
dohe0342 2023-01-03 10:41:03 +09:00
parent 545c3920f1
commit eeb593ae42
2 changed files with 1 additions and 1 deletions

View File

@ -222,7 +222,7 @@ def get_annealed_rate(start, end, curr_step, total_steps):
return end - r * pct_remaining
@register_model("data2vec_audio", dataclass=Data2VecAudioConfig)
@register_model("data2vec_audio_icefall", dataclass=Data2VecAudioConfig)
class Data2VecAudioModel(BaseFairseqModel):
def __init__(self, cfg: Data2VecAudioConfig):
super().__init__()