OcTra / configurations /get_hyperparameters.py
arcan3's picture
adde revision
fc5ed00
raw
history blame
703 Bytes
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'