| --- |
| 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"], |
| ) |
| ``` |
| |
