search

gitmylo / bark-voice-cloning-HuBERT-quantizer

The code for the bark-voicecloning model. Training and inference.
MIT License
671 stars 111 forks source link

simplify model path management #4

Closed rsxdalv closed 1 year ago

rsxdalv commented 1 year ago

This enables hubert_path = HuBERTManager.make_sure_hubert_installed() hubert_model = CustomHubert(checkpoint_path=hubert_path)

tokenizer_path = HuBERTManager.make_sure_tokenizer_installed() tokenizer = CustomTokenizer() tokenizer.load_state_dict(torch.load(tokenizer_path))

gitmylo commented 1 year ago

Good idea. This should simplify it even more.