fixed silly bugs. can work now.

general_dit.py

@@ -390,16 +390,16 @@ class GeneralDIT(nn.Module):
             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.")
 
-        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:

text2world_hf.py

@@ -83,7 +83,7 @@ class DiffusionText2World(PreTrainedModel):
             prompts = read_prompts_from_file(cfg.batch_input_path)
         else:
             # Single prompt case
-            prompts = [{"prompt": cfg.prompt}]
+            prompts = [{"prompt": prompt}]
 
         os.makedirs(cfg.video_save_folder, exist_ok=True)
         for i, input_dict in enumerate(prompts):