from transformers import Wav2Vec2Model, Wav2Vec2ForCTC | |
from .configuration_fake_custom import FakeCustomConfig | |
class FakeCustomModel(Wav2Vec2Model): | |
config_class = FakeCustomConfig | |
class FakeCustomModelForCTC(Wav2Vec2ForCTC): | |
config_class = FakeCustomConfig | |