u

README.md

@@ -12,9 +12,9 @@ tags:
 - zigzag
 
 ---
-# ZigMa: Zigzag Mamba Diffusion Model
+# ZigMa: A DiT-style Zigzag Mamba Diffusion Model
 
-This model represents the official checkpoint of the paper titled "ZigMa: Zigzag Mamba Diffusion Model".
+This model represents the official checkpoint of the paper titled "ZigMa: A DiT-style Zigzag Mamba Diffusion Model".
 
 
 [![Website](doc/badges/badge-website.svg)](https://taohu.me/project_zigma)
@@ -41,7 +41,7 @@ We present ZigMa, a scanning scheme that follows a zigzag pattern, considering b
 
 ```bibtex
 @InProceedings{hu2024zigma,
-      title={ZigMa: Zigzag Mamba Diffusion Model},
+      title={ZigMa: A DiT-style Zigzag Mamba Diffusion Model},
       author={Vincent Tao Hu and Stefan Andreas Baumann and Ming Gui and Olga Grebenkova and Pingchuan Ma and Johannes Fischer and Bjorn Ommer},
       booktitle = {Arxiv},
       year={2024}