ConvTasNet_Lyrics_Causal / load_model.py
groadabike's picture
Upload load_model.py
6ea60c5 verified
from tasnet import ConvTasNetStereo
model_noncausal = ConvTasNetStereo.from_pretrained(
"cadenzachallenge/ConvTasNet_Lyrics_NonCausal"
).cpu()
model_causal = ConvTasNetStereo.from_pretrained(
"cadenzachallenge/ConvTasNet_Lyrics_Causal"
).cpu()