File size: 1,073 Bytes
cee377d
 
 
 
 
 
 
 
 
 
 
40642c4
cee377d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
---
language:
- en
tags:
- text-to-motion
- motion-generation
- retrieval-augmented-generation
---

# ReMoMask-2: Latent Retrieval-Augmented Masked Motion Generation

> Yiran Wang<sup>*</sup>, Zeyu Zhang<sup>*†</sup>, Ling Shao, and Hao Tang<sup>‡</sup>
>
> <sup>*</sup>Equal contribution. <sup>†</sup>Project lead. <sup>‡</sup>Corresponding author.
>
> **arXiv:** Coming soon · **Project Page:** Coming soon

## Inference assets

| Files | Contents |
| --- | --- |
| `net_best_fid_ep0309.tar` | Masked motion generation model |
| `vq.tar` | RVQ model |
| `query_projector.pt` | Text-to-motion query projector |
| `model_opt.txt`, `vq_opt.txt` | Model configurations |
| `mean.npy`, `std.npy` | Motion normalization |
| `encoded_motions.npy`, `encoded_texts_clip.npy` | Retrieval features |
| `motion_ids.npy`, `all_captions.npy` | Retrieval metadata |

## Download

```python
from huggingface_hub import snapshot_download

snapshot_download(
    "yiranranranra/ReMoMask-2",
    local_dir="remomask2",
    allow_patterns=["*.tar", "*.pt", "*.txt", "*.npy"],
)
```