nvidia
/

Nemotron-4-340B-Reward

Model card Files Files and versions Community

zhilinw commited on Jun 18

Commit

4332a35

•

1 Parent(s): 08c5956

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -75,7 +75,7 @@ BF16 Inference:
 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).
-1. Spin up an inference server within the [NeMo Aligner container](https://github.com/NVIDIA/NeMo-Aligner/blob/main/Dockerfile)
 ```
 python /opt/NeMo-Aligner/examples/nlp/gpt/serve_reward_model.py \

 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).
+1. Spin up an inference server within the NeMo container (`docker pull nvcr.io/nvidia/nemo:24.01.framework`)
 ```
 python /opt/NeMo-Aligner/examples/nlp/gpt/serve_reward_model.py \