Missing import for inference

README.md

@@ -47,6 +47,7 @@ import os
 import torch
 from torch import nn
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import snapshot_download
 
 ## Define the reward model function class