Skip to content

Commit d784412

Browse files
authored
fix ckpt-converter readme (#96)
1 parent 7de4c40 commit d784412

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

ckpt_converter/README.md

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,18 @@
1010
- Input checkpoints located at:
1111
- `TASK_DIR/checkpoints/iter_XXXXXXX` (7-digit zero-padded).
1212

13+
How to install anappropriate venv (using uv):
14+
15+
```shell
16+
mkdir cpuenv
17+
cd cpuenv
18+
uv python pin 3.13
19+
uv init
20+
uv venv
21+
uv pip install "torch==2.8.0+cpu" -i https://download.pytorch.org/whl/cpu
22+
uv pip install -r /path/to/ckpt_converter/requirements.txt
23+
```
24+
1325
### Usage
1426
```bash
1527
bash scripts/ckpt_converter/convert_to_hf_cpu.sh \

0 commit comments

Comments
 (0)