Skip to content

Commit 3b729af

Browse files
committed
readd generating samples before training.
1 parent 66c85fe commit 3b729af

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxdiffusion/trainers/wan_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def start_training(self):
146146
# del pipeline.vae
147147

148148
# Generate a sample before training to compare against generated sample after training.
149-
#pretrained_video_path = generate_sample(self.config, pipeline, filename_prefix="pre-training-")
149+
pretrained_video_path = generate_sample(self.config, pipeline, filename_prefix="pre-training-")
150150
mesh = pipeline.mesh
151151
data_iterator = self.load_dataset(mesh)
152152

0 commit comments

Comments
 (0)