Ver código fonte

Fix import order

Lengyue 2 anos atrás
pai
commit
e6126b0ac6
1 arquivos alterados com 2 adições e 2 exclusões
  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 lightning.pytorch.loggers import Logger
 from omegaconf import DictConfig, OmegaConf
 from omegaconf import DictConfig, OmegaConf
 
 
-import fish_speech.utils as utils
-
 os.environ.pop("SLURM_NTASKS", None)
 os.environ.pop("SLURM_NTASKS", None)
 os.environ.pop("SLURM_JOB_NAME", None)
 os.environ.pop("SLURM_JOB_NAME", None)
 os.environ.pop("SLURM_NTASKS_PER_NODE", None)
 os.environ.pop("SLURM_NTASKS_PER_NODE", None)
@@ -25,6 +23,8 @@ torch.backends.cudnn.allow_tf32 = True
 # register eval resolver
 # register eval resolver
 OmegaConf.register_new_resolver("eval", eval)
 OmegaConf.register_new_resolver("eval", eval)
 
 
+import fish_speech.utils as utils
+
 log = utils.RankedLogger(__name__, rank_zero_only=True)
 log = utils.RankedLogger(__name__, rank_zero_only=True)