Update infer_zipvoice.py

infer_zipvoice.py

@@ -245,18 +245,18 @@ model_defaults = {
 device = torch.device("cuda", 0)
 
 print("Loading model...")
-model_config = "ckpt/model.json"
+model_config = "config.json"
 
 with open(model_config, "r") as f:
     model_config = json.load(f)
 
-token_file = "ckpt/tokens.txt"
+token_file = "tokens.txt"
 
 tokenizer = EspeakTokenizer(token_file=token_file, lang="vi")
 
 tokenizer_config = {"vocab_size": tokenizer.vocab_size, "pad_id": tokenizer.pad_id}
 
-model_ckpt = "ckpt/model.pt"
+model_ckpt = "iter-57000-avg-2.pt"
 
 model = ZipVoice(
     **model_config["model"],