Upload predict.py

predict.py

@@ -31,7 +31,7 @@ class Predictor(BasePredictor):
         from bark_hubert_quantizer.customtokenizer import CustomTokenizer
 
         large_quant_model = False  # Use the larger pretrained model
-        device = torch.device('cuda')  # 'cuda', 'cpu', 'cuda:0', 0, -1, torch.device('cuda')
+        device = torch.device('cpu')  # 'cuda', 'cpu', 'cuda:0', 0, -1, torch.device('cuda')
 
         model = ('quantifier_V1_hubert_base_ls960_23.pth', 'tokenizer_large.pth') if large_quant_model else (
             'quantifier_hubert_base_ls960_14.pth', 'tokenizer.pth')