OpenSound commited on
Commit
f725df5
·
verified ·
1 Parent(s): ac91d27

Update inference_scale.py

Browse files
Files changed (1) hide show
  1. inference_scale.py +2 -0
inference_scale.py CHANGED
@@ -17,6 +17,8 @@ import time
17
  @torch.no_grad()
18
  def inference_one_sample(model, model_args, phn2num, text_tokenizer, audio_tokenizer, audio_fn, prompt_text, target_text, mask_interval, cfg_coef, cfg_stride, aug_text, aug_context, use_watermark, tts, device, decode_config):
19
  # phonemize
 
 
20
  text_tokens = [phn2num[phn] for phn in
21
  tokenize_text(
22
  text_tokenizer, text=target_text.strip()
 
17
  @torch.no_grad()
18
  def inference_one_sample(model, model_args, phn2num, text_tokenizer, audio_tokenizer, audio_fn, prompt_text, target_text, mask_interval, cfg_coef, cfg_stride, aug_text, aug_context, use_watermark, tts, device, decode_config):
19
  # phonemize
20
+ print(phn2num)
21
+ print(tokenize_text(text_tokenizer, text=target_text.strip()))
22
  text_tokens = [phn2num[phn] for phn in
23
  tokenize_text(
24
  text_tokenizer, text=target_text.strip()