|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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) |
|
|
|
