from transformers import Wav2Vec2Model, Wav2Vec2ForCTC
from .configuration_fake_custom import FakeCustomConfig

class FakeCustomModel(Wav2Vec2Model):
    config_class = FakeCustomConfig

class FakeCustomModelForCTC(Wav2Vec2ForCTC):
    config_class = FakeCustomConfig