File size: 766 Bytes
3d79eb3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
#
from fairseq2.models.config_loader import StandardModelConfigLoader
from fairseq2.models.loader import StandardModelLoader, load_model
from lcm.models.sonar_normalizer.builder import (
SonarNormalizerConfig,
create_sonar_normalizer,
sonar_normalizer_archs,
)
load_sonar_normalizer_config = StandardModelConfigLoader(
family="sonar_normalizer",
config_kls=SonarNormalizerConfig,
arch_configs=sonar_normalizer_archs,
)
load_sonar_normalizer_model = StandardModelLoader(
config_loader=load_sonar_normalizer_config,
factory=create_sonar_normalizer,
restrict_checkpoints=False,
)
load_model.register("sonar_normalizer", load_sonar_normalizer_model)
|