Lengyue 2 лет назад
Родитель
Сommit
e6126b0ac6
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      fish_speech/train.py

+ 2 - 2
fish_speech/train.py

@@ -9,8 +9,6 @@ from lightning import Callback, LightningDataModule, LightningModule, Trainer
 from lightning.pytorch.loggers import Logger
 from omegaconf import DictConfig, OmegaConf
 
-import fish_speech.utils as utils
-
 os.environ.pop("SLURM_NTASKS", None)
 os.environ.pop("SLURM_JOB_NAME", None)
 os.environ.pop("SLURM_NTASKS_PER_NODE", None)
@@ -25,6 +23,8 @@ torch.backends.cudnn.allow_tf32 = True
 # register eval resolver
 OmegaConf.register_new_resolver("eval", eval)
 
+import fish_speech.utils as utils
+
 log = utils.RankedLogger(__name__, rank_zero_only=True)