OpenSound commited on
Commit
d6547b1
·
verified ·
1 Parent(s): 35576e8

Update inference_scale.py

Browse files
Files changed (1) hide show
  1. inference_scale.py +0 -2
inference_scale.py CHANGED
@@ -17,8 +17,6 @@ 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
- 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()
 
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()