load_model.py

from tasnet import ConvTasNetStereo

model_noncausal = ConvTasNetStereo.from_pretrained(
    "cadenzachallenge/ConvTasNet_Lyrics_NonCausal"
).cpu()

model_causal = ConvTasNetStereo.from_pretrained(
    "cadenzachallenge/ConvTasNet_Lyrics_Causal"
).cpu()