Update RM_demo.py
Browse files- RM_demo.py +1 -1
RM_demo.py
CHANGED
@@ -47,7 +47,7 @@ if __name__ == '__main__':
|
|
47 |
|
48 |
local_model_dir = "Your local model dir"
|
49 |
model_dir = f"{local_model_dir}/Seed-X-RM-7B"
|
50 |
-
prompt = ["Translate the following English sentence into Chinese:\nMay the force be with you <zh>"]
|
51 |
candidate = ["愿原力与你同在","愿力量与你同在"]
|
52 |
model = RewardModel(model_dir)
|
53 |
scores = model.score(prompt, candidate) # output [1.46875, -0.376953125]
|
|
|
47 |
|
48 |
local_model_dir = "Your local model dir"
|
49 |
model_dir = f"{local_model_dir}/Seed-X-RM-7B"
|
50 |
+
prompt = ["Translate the following English sentence into Chinese:\nMay the force be with you <zh>", "Translate the following English sentence into Chinese:\nMay the force be with you <zh>"]
|
51 |
candidate = ["愿原力与你同在","愿力量与你同在"]
|
52 |
model = RewardModel(model_dir)
|
53 |
scores = model.score(prompt, candidate) # output [1.46875, -0.376953125]
|