Add pipeline tag and improve usage documentation

README.md

@@ -1,5 +1,6 @@
 ---
 license: mit
+pipeline_tag: unconditional-image-generation
 tags:
 - diffusion
 - image-generation
@@ -19,7 +20,7 @@ tags:
 
 ## 📄 Abstract
 
-We propose LayerSync, a domain-agnostic approach for improving the generation quality and the training efficiency of diffusion models. Prior studies have highlighted the connection between the quality of generation and the representations learned by diffusion models, showing that external guidance on model intermediate representations accelerates training. We reconceptualize this paradigm by regularizing diffusion models with their own intermediate representations. Building on the observation that representation quality varies across diffusion model layers, we show that the most semantically rich representations can act as an intrinsic guidance for weaker ones, reducing the need for external supervision. Our approach, LayerSync, is a self-sufficient, plug-and-play regularizer term with no overhead on diffusion model training and generalizes beyond the visual domain to other modalities. LayerSync requires no pretrained models nor additional data. We extensively evaluate the method on image generation and demonstrate its applicability to other domains such as audio, video, and human motion generation. LayerSync consistently enhances the generation quality and the training efficiency. For example, it speeds up the training of flow-based transformer by over 8.75x on the ImageNet dataset and improves the generation quality by 23.6%.
+We propose LayerSync, a domain-agnostic approach for improving the generation quality and the training efficiency of diffusion models. Prior studies have highlighted the connection between the quality of generation and the representations learned by diffusion models, showing that external guidance on model intermediate representations accelerates training. We reconceptualize this paradigm by regularizing diffusion models with their own intermediate representations. Our approach, LayerSync, is a self-sufficient, plug-and-play regularizer term with no overhead on diffusion model training and generalizes beyond the visual domain to other modalities.
 
 ---
 
@@ -36,6 +37,14 @@ This repository contains pretrained model checkpoints:
 
 Download the checkpoints and use them with the [LayerSync codebase](https://github.com/vita-epfl/LayerSync).
 
+To generate images using the provided scripts:
+
+```bash
+torchrun --nnodes=1 --nproc_per_node=N sample_ddp.py ODE \
+  --model SiT-XL/2 \
+  --num-fid-samples 50000
+```
+
 ---
 
 ## 📚 Citation