Update README.md
Browse files
README.md
CHANGED
@@ -3,7 +3,9 @@ license: gpl
|
|
3 |
---
|
4 |
|
5 |
usage: https://github.com/CarperAI/trlx/blob/3396bf118869cd823b6909e082caf3810d8989e0/examples/nemo_ilql_inference.py
|
6 |
-
|
|
|
|
|
7 |
```
|
8 |
python trlx/examples/nemo_ilql_inference.py trlx/configs/nemo_configs/megatron_65b.yaml "/path/to/checkpoints"
|
9 |
```
|
|
|
3 |
---
|
4 |
|
5 |
usage: https://github.com/CarperAI/trlx/blob/3396bf118869cd823b6909e082caf3810d8989e0/examples/nemo_ilql_inference.py
|
6 |
+
|
7 |
+
edit above file and below config to have `num_nodes = 2` and `devices = 8`
|
8 |
+
|
9 |
```
|
10 |
python trlx/examples/nemo_ilql_inference.py trlx/configs/nemo_configs/megatron_65b.yaml "/path/to/checkpoints"
|
11 |
```
|