We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 66c85fe commit 3b729afCopy full SHA for 3b729af
1 file changed
src/maxdiffusion/trainers/wan_trainer.py
@@ -146,7 +146,7 @@ def start_training(self):
146
# del pipeline.vae
147
148
# 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-")
+ pretrained_video_path = generate_sample(self.config, pipeline, filename_prefix="pre-training-")
150
mesh = pipeline.mesh
151
data_iterator = self.load_dataset(mesh)
152
0 commit comments