Sync model card with public code release layout
Browse files
README.md
CHANGED
|
@@ -25,7 +25,7 @@ tags:
|
|
| 25 |
|
| 26 |
**Sipeng He · Jialei Chen · Zhen Fang · Dongchun Ren · Feng Zhao**
|
| 27 |
|
| 28 |
-
[Paper](https://arxiv.org/abs/2609.06948) · [Code](https://github.com/Pre-cici/
|
| 29 |
|
| 30 |
</div>
|
| 31 |
|
|
@@ -60,16 +60,17 @@ Download the checkpoint directly into the layout expected by the PRG-Fusion code
|
|
| 60 |
```bash
|
| 61 |
hf download Post-pone/PRG-Fusion \
|
| 62 |
--include '.metadata' '*.distcp' \
|
| 63 |
-
--local-dir checkpoints/
|
| 64 |
```
|
| 65 |
|
| 66 |
The complete checkpoint is approximately 15.95 GB:
|
| 67 |
|
| 68 |
```text
|
| 69 |
-
checkpoints/
|
| 70 |
-
├──
|
| 71 |
-
├──
|
| 72 |
-
|
|
|
|
| 73 |
```
|
| 74 |
|
| 75 |
All three files are required. Keep the original filenames because `.metadata` references the two DCP shards.
|
|
@@ -79,14 +80,15 @@ All three files are required. Keep the original filenames because `.metadata` re
|
|
| 79 |
Clone and install the [PRG-Fusion code](https://github.com/Pre-cici/cosmos) on a Linux NVIDIA GPU host:
|
| 80 |
|
| 81 |
```bash
|
| 82 |
-
git clone https://github.com/Pre-cici/
|
| 83 |
-
cd
|
| 84 |
|
|
|
|
| 85 |
uv sync --extra cu130
|
| 86 |
uv run --extra cu130 python examples/check_environment.py
|
| 87 |
```
|
| 88 |
|
| 89 |
-
Use `cu128` instead of `cu130` on a CUDA 12.8 system. Follow the repository's [checkpoint guide](https://github.com/Pre-cici/
|
| 90 |
|
| 91 |
Prepare an aligned pair of target-trajectory control videos:
|
| 92 |
|
|
@@ -135,6 +137,3 @@ PRG-Fusion is a derivative of NVIDIA Cosmos-Transfer2.5-2B and is released under
|
|
| 135 |
}
|
| 136 |
```
|
| 137 |
|
| 138 |
-
## Acknowledgements
|
| 139 |
-
|
| 140 |
-
**Built on NVIDIA Cosmos.** We thank the NVIDIA Cosmos team and the open-source community for making this work possible.
|
|
|
|
| 25 |
|
| 26 |
**Sipeng He · Jialei Chen · Zhen Fang · Dongchun Ren · Feng Zhao**
|
| 27 |
|
| 28 |
+
[Paper](https://arxiv.org/abs/2609.06948) · [Code](https://github.com/Pre-cici/PRG-Fusion)
|
| 29 |
|
| 30 |
</div>
|
| 31 |
|
|
|
|
| 60 |
```bash
|
| 61 |
hf download Post-pone/PRG-Fusion \
|
| 62 |
--include '.metadata' '*.distcp' \
|
| 63 |
+
--local-dir checkpoints/PRG-Fusion/model
|
| 64 |
```
|
| 65 |
|
| 66 |
The complete checkpoint is approximately 15.95 GB:
|
| 67 |
|
| 68 |
```text
|
| 69 |
+
checkpoints/PRG-Fusion/
|
| 70 |
+
├── model/
|
| 71 |
+
│ ├── .metadata
|
| 72 |
+
│ ├── __0_0.distcp
|
| 73 |
+
│ └── __1_0.distcp
|
| 74 |
```
|
| 75 |
|
| 76 |
All three files are required. Keep the original filenames because `.metadata` references the two DCP shards.
|
|
|
|
| 80 |
Clone and install the [PRG-Fusion code](https://github.com/Pre-cici/cosmos) on a Linux NVIDIA GPU host:
|
| 81 |
|
| 82 |
```bash
|
| 83 |
+
git clone https://github.com/Pre-cici/PRG-Fusion.git
|
| 84 |
+
cd PRG-Fusion
|
| 85 |
|
| 86 |
+
uv venv --python 3.10
|
| 87 |
uv sync --extra cu130
|
| 88 |
uv run --extra cu130 python examples/check_environment.py
|
| 89 |
```
|
| 90 |
|
| 91 |
+
Use `cu128` instead of `cu130` on a CUDA 12.8 system. Follow the repository's [checkpoint guide](https://github.com/Pre-cici/PRG-Fusion/blob/main/docs/checkpoints.md) to download the remaining Cosmos tokenizer and text-conditioning components.
|
| 92 |
|
| 93 |
Prepare an aligned pair of target-trajectory control videos:
|
| 94 |
|
|
|
|
| 137 |
}
|
| 138 |
```
|
| 139 |
|
|
|
|
|
|
|
|
|