Update README.md
Browse files
README.md
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
<div align="center">
|
2 |
|
3 |
|
4 |
-
<img src="icon.jpg" width="250"/>
|
5 |
|
6 |
<h2><center>Tora: Trajectory-oriented Diffusion Transformer for Video Generation</h2>
|
7 |
|
@@ -17,8 +16,12 @@ Zhenghao Zhang\*, Junchao Liao\*, Menghao Li, Zuozhuo Dai, Bingxue Qiu, Siyu Zhu
|
|
17 |
|
18 |
<a href='https://modelscope.cn/models/xiaoche/Tora'><img src='https://img.shields.io/badge/🤖_ModelScope-weights-%23654dfc'></a>
|
19 |
<a href='https://huggingface.co/Le0jc/Tora'><img src='https://img.shields.io/badge/🤗_HuggingFace-weights-%23ff9e0e'></a>
|
|
|
20 |
</div>
|
21 |
|
|
|
|
|
|
|
22 |
|
23 |
## 💡 Abstract
|
24 |
|
@@ -26,6 +29,8 @@ Recent advancements in Diffusion Transformer (DiT) have demonstrated remarkable
|
|
26 |
|
27 |
## 📣 Updates
|
28 |
|
|
|
|
|
29 |
- `2024/10/31` Model weights uploaded to [HuggingFace](https://huggingface.co/Le0jc/Tora). We also provided an English demo on [ModelScope](https://www.modelscope.cn/studios/Alibaba_Research_Intelligence_Computing/Tora_En).
|
30 |
- `2024/10/23` 🔥🔥Our [ModelScope Demo](https://www.modelscope.cn/studios/xiaoche/Tora) is launched. Welcome to try it out! We also upload the model weights to [ModelScope](https://www.modelscope.cn/models/xiaoche/Tora).
|
31 |
- `2024/10/21` Thanks to [@kijai](https://github.com/kijai) for supporting Tora in ComfyUI! [Link](https://github.com/kijai/ComfyUI-CogVideoXWrapper)
|
@@ -33,15 +38,6 @@ Recent advancements in Diffusion Transformer (DiT) have demonstrated remarkable
|
|
33 |
- `2024/08/27` We released our v2 paper including appendix.
|
34 |
- `2024/07/31` We submitted our paper on arXiv and released our project page.
|
35 |
|
36 |
-
## 📑 Table of Contents
|
37 |
-
|
38 |
-
- [Showcases](#%EF%B8%8F-showcases)
|
39 |
-
- [Model Weights](#-model-weights)
|
40 |
-
- [Inference](#-inference)
|
41 |
-
- [Acknowledgements](#-acknowledgements)
|
42 |
-
- [Our previous work](#-our-previous-work)
|
43 |
-
- [Citation](#-citation)
|
44 |
-
|
45 |
## 🎞️ Showcases
|
46 |
|
47 |
https://github.com/user-attachments/assets/949d5e99-18c9-49d6-b669-9003ccd44bf1
|
@@ -52,80 +48,6 @@ https://github.com/user-attachments/assets/4026c23d-229d-45d7-b5be-6f3eb9e4fd50
|
|
52 |
|
53 |
All videos are available in this [Link](https://cloudbook-public-daily.oss-cn-hangzhou.aliyuncs.com/Tora_t2v/showcases.zip)
|
54 |
|
55 |
-
|
56 |
-
## 📦 Model Weights
|
57 |
-
|
58 |
-
### Folder Structure
|
59 |
-
|
60 |
-
```
|
61 |
-
Tora
|
62 |
-
└── sat
|
63 |
-
└── ckpts
|
64 |
-
├── t5-v1_1-xxl
|
65 |
-
│ ├── model-00001-of-00002.safetensors
|
66 |
-
│ └── ...
|
67 |
-
├── vae
|
68 |
-
│ └── 3d-vae.pt
|
69 |
-
└── tora
|
70 |
-
└── t2v
|
71 |
-
└── mp_rank_00_model_states.pt
|
72 |
-
```
|
73 |
-
|
74 |
-
### Download Links
|
75 |
-
*Note: Downloading the `tora` weights requires following the [CogVideoX License](CogVideoX_LICENSE).* You can choose one of the following options: HuggingFace, ModelScope, or native links.
|
76 |
-
After downloading the model weights, you can put them in the `Tora/sat/ckpts` folder.
|
77 |
-
|
78 |
-
#### HuggingFace
|
79 |
-
|
80 |
-
```bash
|
81 |
-
# This can be faster
|
82 |
-
pip install "huggingface_hub[hf_transfer]"
|
83 |
-
HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download Le0jc/Tora --local-dir ckpts
|
84 |
-
```
|
85 |
-
|
86 |
-
or
|
87 |
-
|
88 |
-
```bash
|
89 |
-
# use git
|
90 |
-
git lfs install
|
91 |
-
git clone https://huggingface.co/Le0jc/Tora
|
92 |
-
```
|
93 |
-
#### ModelScope
|
94 |
-
- SDK
|
95 |
-
```bash
|
96 |
-
from modelscope import snapshot_download
|
97 |
-
model_dir = snapshot_download('xiaoche/Tora')
|
98 |
-
```
|
99 |
-
|
100 |
-
- Git
|
101 |
-
```bash
|
102 |
-
git clone https://www.modelscope.cn/xiaoche/Tora.git
|
103 |
-
```
|
104 |
-
|
105 |
-
#### Native
|
106 |
-
|
107 |
-
- Download the VAE and T5 model following [CogVideo](https://github.com/THUDM/CogVideo/blob/main/sat/README.md#2-download-model-weights):
|
108 |
-
- VAE: https://cloud.tsinghua.edu.cn/f/fdba7608a49c463ba754/?dl=1
|
109 |
-
- T5: [text_encoder](https://huggingface.co/THUDM/CogVideoX-2b/tree/main/text_encoder), [tokenizer](https://huggingface.co/THUDM/CogVideoX-2b/tree/main/tokenizer)
|
110 |
-
- Tora t2v model weights: [Link](https://cloudbook-public-daily.oss-cn-hangzhou.aliyuncs.com/Tora_t2v/mp_rank_00_model_states.pt). Downloading this weight requires following the [CogVideoX License](CogVideoX_LICENSE).
|
111 |
-
|
112 |
-
|
113 |
-
|
114 |
-
## 🔄 Inference
|
115 |
-
|
116 |
-
please refer to our [Github](https://github.com/alibaba/Tora) or [modelscope online demo](https://www.modelscope.cn/studios/Alibaba_Research_Intelligence_Computing/Tora_En)
|
117 |
-
|
118 |
-
### Recommendations for Text Prompts
|
119 |
-
|
120 |
-
For text prompts, we highly recommend using GPT-4 to enhance the details. Simple prompts may negatively impact both visual quality and motion control effectiveness.
|
121 |
-
|
122 |
-
You can refer to the following resources for guidance:
|
123 |
-
|
124 |
-
- [CogVideoX Documentation](https://github.com/THUDM/CogVideo/blob/main/inference/convert_demo.py)
|
125 |
-
- [OpenSora Scripts](https://github.com/hpcaitech/Open-Sora/blob/main/scripts/inference.py)
|
126 |
-
|
127 |
-
|
128 |
-
|
129 |
## 🤝 Acknowledgements
|
130 |
|
131 |
We would like to express our gratitude to the following open-source projects that have been instrumental in the development of our project:
|
|
|
1 |
<div align="center">
|
2 |
|
3 |
|
|
|
4 |
|
5 |
<h2><center>Tora: Trajectory-oriented Diffusion Transformer for Video Generation</h2>
|
6 |
|
|
|
16 |
|
17 |
<a href='https://modelscope.cn/models/xiaoche/Tora'><img src='https://img.shields.io/badge/🤖_ModelScope-weights-%23654dfc'></a>
|
18 |
<a href='https://huggingface.co/Le0jc/Tora'><img src='https://img.shields.io/badge/🤗_HuggingFace-weights-%23ff9e0e'></a>
|
19 |
+
|
20 |
</div>
|
21 |
|
22 |
+
This is the official repository for paper "Tora: Trajectory-oriented Diffusion Transformer for Video Generation".
|
23 |
+
|
24 |
+
## Please visit our [Github repo](https://github.com/alibaba/Tora) for more details.
|
25 |
|
26 |
## 💡 Abstract
|
27 |
|
|
|
29 |
|
30 |
## 📣 Updates
|
31 |
|
32 |
+
- `2024/12/09` 🔥🔥Diffusers version of Tora and the corresponding model weights are released. Inference VRAM requirements are reduced to around 5 GiB. Please refer to [this](diffusers-version/README.md) for details.
|
33 |
+
- `2024/11/25` 🔥Text-to-Video training code released.
|
34 |
- `2024/10/31` Model weights uploaded to [HuggingFace](https://huggingface.co/Le0jc/Tora). We also provided an English demo on [ModelScope](https://www.modelscope.cn/studios/Alibaba_Research_Intelligence_Computing/Tora_En).
|
35 |
- `2024/10/23` 🔥🔥Our [ModelScope Demo](https://www.modelscope.cn/studios/xiaoche/Tora) is launched. Welcome to try it out! We also upload the model weights to [ModelScope](https://www.modelscope.cn/models/xiaoche/Tora).
|
36 |
- `2024/10/21` Thanks to [@kijai](https://github.com/kijai) for supporting Tora in ComfyUI! [Link](https://github.com/kijai/ComfyUI-CogVideoXWrapper)
|
|
|
38 |
- `2024/08/27` We released our v2 paper including appendix.
|
39 |
- `2024/07/31` We submitted our paper on arXiv and released our project page.
|
40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
41 |
## 🎞️ Showcases
|
42 |
|
43 |
https://github.com/user-attachments/assets/949d5e99-18c9-49d6-b669-9003ccd44bf1
|
|
|
48 |
|
49 |
All videos are available in this [Link](https://cloudbook-public-daily.oss-cn-hangzhou.aliyuncs.com/Tora_t2v/showcases.zip)
|
50 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
## 🤝 Acknowledgements
|
52 |
|
53 |
We would like to express our gratitude to the following open-source projects that have been instrumental in the development of our project:
|