mirror of
https://github.com/FunAudioLLM/CosyVoice.git
synced 2026-02-04 17:39:25 +08:00
fix bug
This commit is contained in:
@@ -146,7 +146,7 @@ def Dataset(data_list_file,
|
|||||||
partition=partition)
|
partition=partition)
|
||||||
# map partial arg to padding func
|
# map partial arg to padding func
|
||||||
for i in range(1, len(data_pipeline)):
|
for i in range(1, len(data_pipeline)):
|
||||||
if data_pipeline[i].func.__name__ == 'compute_fbank':
|
if data_pipeline[i].func.__name__ == 'compute_fbank' and gan is True:
|
||||||
data_pipeline[i] = partial(data_pipeline[i], token_mel_ratio=0)
|
data_pipeline[i] = partial(data_pipeline[i], token_mel_ratio=0)
|
||||||
if data_pipeline[i].func.__name__ == 'padding':
|
if data_pipeline[i].func.__name__ == 'padding':
|
||||||
data_pipeline[i] = partial(data_pipeline[i], gan=gan, dpo=dpo)
|
data_pipeline[i] = partial(data_pipeline[i], gan=gan, dpo=dpo)
|
||||||
|
|||||||
Reference in New Issue
Block a user