Post-pone commited on
Commit
a3455c4
·
verified ·
1 Parent(s): 19af7dc

Sync model card with public code release layout

Browse files
Files changed (1) hide show
  1. README.md +11 -12
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/cosmos)
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/omnire/iter_000001000/model
64
  ```
65
 
66
  The complete checkpoint is approximately 15.95 GB:
67
 
68
  ```text
69
- checkpoints/omnire/iter_000001000/model/
70
- ├── .metadata
71
- ├── __0_0.distcp
72
- └── __1_0.distcp
 
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/cosmos.git
83
- cd cosmos
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/cosmos/blob/main/docs/checkpoints.md) to download the remaining Cosmos tokenizer and text-conditioning components.
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