Add link to Github repository

README.md

@@ -1,12 +1,13 @@
 ---
-license: apache-2.0
 base_model:
 - THUDM/CogVideoX-2b
-pipeline_tag: text-to-video
 library_name: diffusers
+license: apache-2.0
+pipeline_tag: text-to-video
 tags:
 - lora
 ---
+
 # TDM: Learning Few-Step Diffusion Models by Trajectory Distribution Matching
 <div style="text-align: center;">
   <a href="https://tdm-t2x.github.io/"><img src="https://img.shields.io/static/v1?label=Project%20Page&message=Github-Page&color=blue&logo=github-pages" style="display: inline;"></a> &ensp;
@@ -15,6 +16,7 @@ tags:
 
 This is the Official Repository of "[Learning Few-Step Diffusion Models by Trajectory Distribution Matching](https://arxiv.org/abs/2503.06674)", by *Yihong Luo, Tianyang Hu, Jiacheng Sun, Yujun Cai, Jing Tang*.
 
+Github repository: https://github.com/Luo-Yihong/TDM
 
 ## User Study Time!
 ![user_study](user_study.jpg)
@@ -136,8 +138,7 @@ generator = torch.manual_seed(8888)
 frames = pipe(prompt, guidance_scale=1, 
               num_inference_steps=4, 
               num_frames=49,
-              generator = generator,
-              use_dynamic_cfg=True).frames[0]
+              generator = generator).frames[0]
 export_to_video(frames, "output-TDM.mp4", fps=8)
 ```
 ## 🔥 Pre-trained Models
@@ -146,6 +147,9 @@ We release a bucket of TDM-LoRA. Please enjoy it!
 - [TDM-CogVideoX-2B-LoRA](https://huggingface.co/Luo-Yihong/TDM_CogVideoX-2B_LoRA)
 - [TDM-Dreamshaper-LoRA](https://huggingface.co/Luo-Yihong/TDM_dreamshaper_LoRA)
   
+## 🔥 TO-DO
+- Release the training code.
+
 
 ## Contact