Spaces:
Running
Running
| 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' |