Hugging Face's logo

Spaces:
GitMylo
/
bark-voice-cloning
Running

App Files Community
6
Mylo commited on
Commit
162810d
•
1 Parent(s): b8153de

map to cpu

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -13,7 +13,10 @@ from encodec.utils import convert_audio
13
 
14
 
15
  hubert_model = CustomHubert(HuBERTManager.make_sure_hubert_installed())
16
- tokenizer_model = CustomTokenizer.load_from_checkpoint(HuBERTManager.make_sure_tokenizer_installed(model='quantifier_V1_hubert_base_ls960_23.pth'))
 
 
 
17
  encodec_model = EncodecModel.encodec_model_24khz()
18
 
19
 
 
13
 
14
 
15
  hubert_model = CustomHubert(HuBERTManager.make_sure_hubert_installed())
16
+ tokenizer_model = CustomTokenizer.load_from_checkpoint(
17
+ HuBERTManager.make_sure_tokenizer_installed(model='quantifier_V1_hubert_base_ls960_23.pth'),
18
+ map_location=torch.device('cpu')
19
+ )
20
  encodec_model = EncodecModel.encodec_model_24khz()
21
 
22