Gregniuki commited on
Commit
71de715
1 Parent(s): 8fa3da0

Update model/trainer.py

Browse files
Files changed (1) hide show
  1. model/trainer.py +3 -3
model/trainer.py CHANGED
@@ -14,9 +14,9 @@ from torch.optim.lr_scheduler import LinearLR, SequentialLR
14
  from torch.utils.data import DataLoader, Dataset, SequentialSampler
15
  from tqdm import tqdm
16
 
17
- from f5_tts.model import CFM
18
- from f5_tts.model.dataset import DynamicBatchSampler, collate_fn
19
- from f5_tts.model.utils import default, exists
20
 
21
  # trainer
22
 
 
14
  from torch.utils.data import DataLoader, Dataset, SequentialSampler
15
  from tqdm import tqdm
16
 
17
+ from model import CFM
18
+ from model.dataset import DynamicBatchSampler, collate_fn
19
+ from model.utils import default, exists
20
 
21
  # trainer
22