tiamojames commited on
Commit
a26eb3b
·
verified ·
1 Parent(s): de9311d

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. soulxpodcast/utils/dataloader.py +1 -1
app.py CHANGED
@@ -287,6 +287,7 @@ def process_single(target_text_list, prompt_wav_list, prompt_text_list, use_dial
287
  "use_dialect_prompt": use_dialect_prompt,
288
  }
289
  if use_dialect_prompt:
 
290
  processed_data.update({
291
  "dialect_prompt_text_tokens_for_llm": data["dialect_prompt_text_tokens"],
292
  "dialect_prefix": data["dialect_prefix"],
 
287
  "use_dialect_prompt": use_dialect_prompt,
288
  }
289
  if use_dialect_prompt:
290
+ print(f"dialect_prompt_text_tokens_for_llm: {data['dialect_prompt_text_tokens']}")
291
  processed_data.update({
292
  "dialect_prompt_text_tokens_for_llm": data["dialect_prompt_text_tokens"],
293
  "dialect_prefix": data["dialect_prefix"],
soulxpodcast/utils/dataloader.py CHANGED
@@ -143,7 +143,7 @@ class PodcastDataset(Dataset):
143
  "spk_emb": spk_emb_list, "mel": mel_list, "mel_len": mel_len_list, "log_mel": log_mel_list, "info": data,
144
  }
145
  if use_dialect_prompt:
146
- import pdb; pdb.set_trace()
147
  print(f"dialect_prompt_text_ids: {dialect_prompt_text_ids}")
148
  print(f"dialect_prompt_text: {dialect_prompt_text}")
149
  item.update({
 
143
  "spk_emb": spk_emb_list, "mel": mel_list, "mel_len": mel_len_list, "log_mel": log_mel_list, "info": data,
144
  }
145
  if use_dialect_prompt:
146
+ # import pdb; pdb.set_trace()
147
  print(f"dialect_prompt_text_ids: {dialect_prompt_text_ids}")
148
  print(f"dialect_prompt_text: {dialect_prompt_text}")
149
  item.update({