import torch

class hyperparameterConfig:
    def __init__(self):
        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

        self.stt_model  = "large-v2"
        self.nllb_model = '1.3B'

        # text to speech model
        self.text2speech_model   = 'model_weights/voiceover/freevc-24.pth'
        self.text2speech_config  = 'model_weights/voiceover/freevc-24.json'
        self.text2speech_encoder = 'model_weights/voice_enhance/ckpt/pretrained_bak_5805000.pt'

        # voice enhancing model
        self.voice_enhacing_model = 'model_weights/voice_enhance'

        # loading the wavlm model
        self.wavlm_model = 'model_weights/wavlm_models/WavLM-Large.pt'