remove space

susanbao · susanbao · commit 6e5aee90b05c · 2025-10-01T16:58:56.000Z
diff --git a/src/maxdiffusion/trainers/wan_trainer.py b/src/maxdiffusion/trainers/wan_trainer.py
@@ -176,7 +176,6 @@ def load_dataset(self, mesh, is_training=True):
       raise ValueError(
           "Wan 2.1 training only supports config.dataset_type set to tfrecords and config.cache_latents_text_encoder_outputs set to True"
       )
-    
     feature_description = {
         "latents": tf.io.FixedLenFeature([], tf.string),
         "encoder_hidden_states": tf.io.FixedLenFeature([], tf.string),

Original file line number	Diff line number	Diff line change
`@@ -176,7 +176,6 @@ def load_dataset(self, mesh, is_training=True):`
`176`	`176`	`raise ValueError(`
`177`	`177`	`"Wan 2.1 training only supports config.dataset_type set to tfrecords and config.cache_latents_text_encoder_outputs set to True"`
`178`	`178`	`)`
`179`		`-`
`180`	`179`	`feature_description = {`
`181`	`180`	`"latents": tf.io.FixedLenFeature([], tf.string),`
`182`	`181`	`"encoder_hidden_states": tf.io.FixedLenFeature([], tf.string),`