Update README.md
Browse files
README.md
CHANGED
|
@@ -26,9 +26,9 @@ This model was converted from JAX to PyTorch using the OpenPI conversion script:
|
|
| 26 |
|
| 27 |
```bash
|
| 28 |
python examples/convert_jax_model_to_pytorch.py \
|
| 29 |
-
--checkpoint_dir /
|
| 30 |
--config_name pi0_base \
|
| 31 |
-
--output_path /
|
| 32 |
--precision float32
|
| 33 |
```
|
| 34 |
|
|
|
|
| 26 |
|
| 27 |
```bash
|
| 28 |
python examples/convert_jax_model_to_pytorch.py \
|
| 29 |
+
--checkpoint_dir /pi0_base \
|
| 30 |
--config_name pi0_base \
|
| 31 |
+
--output_path /pi0_base/pytorch/fp32/ \
|
| 32 |
--precision float32
|
| 33 |
```
|
| 34 |
|