Upload predict.py

predict.py

@@ -30,7 +30,7 @@ class Predictor(BasePredictor):
         from bark_hubert_quantizer.pre_kmeans_hubert import CustomHubert
         from bark_hubert_quantizer.customtokenizer import CustomTokenizer
 
-        large_quant_model = False  # Use the larger pretrained model
+        large_quant_model = True  # Use the larger pretrained model
         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 (