xMerge branch 'main' of https://github.com/jzhang38/FastVideo-OSP into peiyuan

2024-12-02 22:31:48 +04:00
parent 74d6fc92a9 daaefc5f63
commit 5325aa2cd2
1 changed files with 1 additions and 1 deletions
@@ -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,