File size: 498 Bytes
b038b10 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# modeling_usad.py
from transformers import PreTrainedModel
from .configuration_usad import USADConfig
from .usad_model import UsadModel as model
class USADModel(PreTrainedModel):
config_class = USADConfig
def __init__(self, config: USADConfig):
super().__init__(config)
self.model = model(config)
def forward(self, *args, **kwargs):
return self.model(*args, **kwargs)
def load_audio(self, audio_path):
return self.model.load_audio(audio_path)
|