Update README.md
Browse files
README.md
CHANGED
@@ -75,7 +75,7 @@ BF16 Inference:
|
|
75 |
|
76 |
You can use the model with [NeMo Aligner](https://github.com/NVIDIA/NeMo-Aligner) following [SteerLM training user guide](https://docs.nvidia.com/nemo-framework/user-guide/latest/modelalignment/steerlm.html).
|
77 |
|
78 |
-
1. Spin up an inference server within the
|
79 |
|
80 |
```
|
81 |
python /opt/NeMo-Aligner/examples/nlp/gpt/serve_reward_model.py \
|
|
|
75 |
|
76 |
You can use the model with [NeMo Aligner](https://github.com/NVIDIA/NeMo-Aligner) following [SteerLM training user guide](https://docs.nvidia.com/nemo-framework/user-guide/latest/modelalignment/steerlm.html).
|
77 |
|
78 |
+
1. Spin up an inference server within the NeMo container (`docker pull nvcr.io/nvidia/nemo:24.01.framework`)
|
79 |
|
80 |
```
|
81 |
python /opt/NeMo-Aligner/examples/nlp/gpt/serve_reward_model.py \
|