Umean commited on
Commit
9b727ea
·
verified ·
1 Parent(s): 73c5b3c

Update README_zh-CN.md

Browse files
Files changed (1) hide show
  1. README_zh-CN.md +1 -0
README_zh-CN.md CHANGED
@@ -106,6 +106,7 @@ batch = tokenizer(encoded_data, return_tensors='pt', padding=True).to('cuda')
106
  outputs = model(**batch)
107
  rewards = outputs[0].squeeze(-1).cpu().tolist()
108
  print(rewards)
 
109
  ```
110
 
111
  ### 使用 lmdeploy 进行推理
 
106
  outputs = model(**batch)
107
  rewards = outputs[0].squeeze(-1).cpu().tolist()
108
  print(rewards)
109
+ # [-0.5702977776527405, -11.030370712280273] for previous example data
110
  ```
111
 
112
  ### 使用 lmdeploy 进行推理