Lengyue před 2 roky
rodič
revize
e6126b0ac6
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  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)