sob111 commited on
Commit
ecaba54
verified
1 Parent(s): dac5923

Update train_gpt_xtts.py

Browse files
Files changed (1) hide show
  1. train_gpt_xtts.py +3 -1
train_gpt_xtts.py CHANGED
@@ -1,8 +1,10 @@
1
  import os
2
  import torch
3
 
 
4
  from TTS.tts.configs.xtts_config import XttsConfig
5
- torch.serialization.add_safe_globals([XttsConfig])
 
6
 
7
  from trainer import Trainer, TrainerArgs
8
 
 
1
  import os
2
  import torch
3
 
4
+ # 馃敀 Permitir deserializaci贸n segura de configuraciones XTTS
5
  from TTS.tts.configs.xtts_config import XttsConfig
6
+ from TTS.tts.models.xtts import XttsAudioConfig
7
+ torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig])
8
 
9
  from trainer import Trainer, TrainerArgs
10