ASLP-lab commited on
Commit
beb10c1
·
verified ·
1 Parent(s): 39080ac

Update diffrhythm2/utils.py

Browse files
Files changed (1) hide show
  1. diffrhythm2/utils.py +1 -1
diffrhythm2/utils.py CHANGED
@@ -128,7 +128,7 @@ def parse_lyrics(lrc_tokenizer, lyrics: str):
128
  tokens = tokens + [STRUCT_INFO['[stop]']]
129
  lyrics_with_time.append(tokens)
130
  if len(lyrics_with_time) != 0 and not get_start:
131
- lyrics_with_time = [STRUCT_INFO['[start]'], STRUCT_INFO['[stop]']] + lyrics_with_time
132
  return lyrics_with_time
133
 
134
  @torch.no_grad()
 
128
  tokens = tokens + [STRUCT_INFO['[stop]']]
129
  lyrics_with_time.append(tokens)
130
  if len(lyrics_with_time) != 0 and not get_start:
131
+ lyrics_with_time = [[STRUCT_INFO['[start]'], STRUCT_INFO['[stop]']]] + lyrics_with_time
132
  return lyrics_with_time
133
 
134
  @torch.no_grad()