Update README.md

README.md

@@ -10,6 +10,10 @@ tags:
   - wan2.2
 ---
 
+<p align="center">
+  <img src="https://raw.githubusercontent.com/wileewang/LongLive2.0/release-clean-merge/assets/longlive2/logo.png" alt="LongLive2.0 logo" width="100%">
+</p>
+
 # LongLive2.0 5B Checkpoints
 
 This repository hosts temporary LongLive2.0 5B checkpoints for inference with
@@ -91,12 +95,14 @@ If the LoRA checkpoint is not used, remove the `adapter` section and leave
 
 ## Prompt Folder
 
-`data.data_path` can be either:
+`data.data_path` is passed to `MultiTextConcatDataset` in `inference.py`. It can
+be either:
 
 - a `.txt` file, where each line is one single-shot prompt; or
 - a directory of multi-shot prompt folders.
 
-Example multi-shot prompt folder:
+For a directory input, the code supports both of the following layouts. The
+direct caption-root layout is the simplest:
 
 ```text
 inference_prompts/
@@ -107,6 +113,18 @@ inference_prompts/
     shot_durations.txt
 ```
 
+It also supports a dataset root with an outer `caption/` folder:
+
+```text
+inference_prompts/
+  caption/
+    robot_lab_demo/
+      0.json
+      1.json
+      2.json
+      shot_durations.txt
+```
+
 Each JSON file contains:
 
 ```json
@@ -148,3 +166,16 @@ Outputs are written to `output_folder`.
 - `inference.multi_shot_rope_offset` controls the multi-shot RoPE offset.
 - For NVFP4 inference, use the separate NVFP4 config and setup instructions in
   the LongLive2.0 documentation.
+
+## Citation
+
+Citation will be updated after the paper is released.
+
+```bibtex
+@article{longlive2,
+  title   = {LongLive2.0: An NVFP4 Parallel Infrastructure for Long Video Generation},
+  author  = {TODO},
+  journal = {TODO},
+  year    = {2026}
+}
+```