Update README.md

README.md

@@ -58,7 +58,7 @@ python inference.py \
 ## Usage Example
 
 This example shows a typical workflow for a **single user**:
-1) encode text pairs with Skywork-Reward-V2-Llama-3.1-8B into embeddings,
+1) encode text pairs with Skywork/Skywork-Reward-Llama-3.1-8B-v0.2 into embeddings,
 2) adapt the MRM on the user's few-shot examples (update `shared_weight` only),
 3) run inference on new pairs for that same user.
 
@@ -107,7 +107,7 @@ def infer_on_pairs(model, ch, rj, device="cuda"):
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-MODEL_PATH = "Skywork/Skywork-Reward-V2-Llama-3.1-8B"
+MODEL_PATH = "Skywork/Skywork-Reward-Llama-3.1-8B-v0.2"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
 llm = AutoModelForSequenceClassification.from_pretrained(
     MODEL_PATH, num_labels=1, torch_dtype=torch.bfloat16, device_map=device