Update README_zh-CN.md
Browse files- 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 进行推理
|