Update README.md

README.md

@@ -33,6 +33,8 @@ We evaluate GRM on the [reward model benchmark](https://huggingface.co/spaces/al
 ## Usage
 **Note 1: Please download the `model.py` file from this repository to ensure the structure is loaded correctly and verify that the `v_head` is properly initialized.**
 
+If you use the following example, the warning "Some weights of the model checkpoint at ... were not used when initializing LlamaForCausalLM" can be just omitted. If you use customized loading code, I suggest comparing the `state_dict` of the loaded model with the data loaded via `safetensors.safe_open(xx.safetensors)` or `torch.load(xx.bin)`. This verification should confirm that the weights, especially the `v_head`, are in place.
+
 **Note 2: loading llama3 model into 8 bit could lead to performance degradation.**
 ```
 import torch