Hey! Could you also release weights for CTC and SV models for distillation?
if ctc_path != "":
model = ConformerCTC(vocab_size=real_unet.vocab_size, mel_dim=real_unet.mel_dim, num_heads=8, d_hid=512, nlayers=6)
self.ctc_model = model.eval()
self.ctc_model.requires_grad_(False)
self.ctc_model.load_state_dict(torch.load(ctc_path, weights_only=True, map_location='cpu')['model_state_dict'])
if sv_path_en != "":
model = ECAPA_TDNN()
self.sv_model_en = model.eval()
self.sv_model_en.requires_grad_(False)
self.sv_model_en.load_state_dict(torch.load(sv_path, weights_only=True, map_location='cpu')['model_state_dict'])
if sv_path_zh != "":
model = ECAPA_TDNN()
self.sv_model_zh = model.eval()
self.sv_model_zh.requires_grad_(False)
self.sv_model_zh.load_state_dict(torch.load(sv_path_zh, weights_only=True, map_location='cpu')['model_state_dict'])
Hey! Could you also release weights for CTC and SV models for distillation?