fix finetune code bug

2024-12-02 11:45:04 +00:00
parent 3eee42e917
commit daaefc5f63
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,