Fixed log in dit

general_dit_video_conditioned.py

@@ -155,11 +155,11 @@ class VideoExtendGeneralDIT(GeneralDIT):
             latent_condition_sigma=latent_condition_sigma,
         )
         # logging affline scale information
-        affline_scale_log.info = {}
+        affline_scale_log_info = {}
 
         timesteps_B_D, adaln_lora_B_3D = self.t_embedder(timesteps.flatten())
         affline_emb_B_D = timesteps_B_D
-        affline_scale_log.info["timesteps_B_D"] = timesteps_B_D.detach()
+        affline_scale_log_info["timesteps_B_D"] = timesteps_B_D.detach()
 
         if scalar_feature is not None:
             raise NotImplementedError("Scalar feature is not implemented yet.")
@@ -173,7 +173,7 @@ class VideoExtendGeneralDIT(GeneralDIT):
 
             affline_augment_sigma_emb_B_D, _ = self.augment_sigma_embedder(condition_video_augment_sigma.flatten())
             affline_emb_B_D = affline_emb_B_D + affline_augment_sigma_emb_B_D
-        affline_scale_log.info["affline_emb_B_D"] = affline_emb_B_D.detach()
+        affline_scale_log_info["affline_emb_B_D"] = affline_emb_B_D.detach()
         affline_emb_B_D = self.affline_norm(affline_emb_B_D)
 
         if self.use_cross_attn_mask: