From daaefc5f63e2cf2263ce0584fb63240d21cf91ea Mon Sep 17 00:00:00 2001 From: haoanyscale Date: Mon, 2 Dec 2024 11:45:04 +0000 Subject: [PATCH] fix finetune code bug --- fastvideo/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastvideo/train.py b/fastvideo/train.py index 5c807512b..6ab95a93a 100644 --- a/fastvideo/train.py +++ b/fastvideo/train.py @@ -249,7 +249,7 @@ def main(args): last_epoch=init_steps - 1, ) - train_dataset = LatentDataset(args.data_json_path, args.num_latent_t, args.cfg, args.uncond_prompt_dir) + train_dataset = LatentDataset(args.data_json_path, args.num_latent_t, args.cfg) sampler = LengthGroupedSampler( args.train_batch_size, rank=rank,